WSL Ubuntu安装TensorFlow-GPU、PyTorch-GPU

在Windows 11的WSL Ubuntu中安装TensorFlow-GPU、PyTorch-GPU


0、WSL Ubuntu安装

  • 在Windows 11的商店中下载即可,此处以Ubuntu22.04.3为例

1、CUDA Toolkit安装

  • 参考公孙启 的文章Windows11 + WSL Ubuntu + Pycharm + Conda for deeplearning
  • 前往nVidia官网下载CUDA Toolkit,这里以11.8为例,因为PyTorch目前支持11.8和12.1两个版本的CUDA
  • 下载时注意选择WSL-Ubuntu,根据网页下方提供的命令依次输入即可完成安装
  • 配置环境变量
    • sudo vi ~/.bashrc

    • 添加以下内容,

      bash 复制代码
      export PATH=/usr/local/cuda-11.8/bin${PATH:+:${PATH}}
      export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
    • 更新环境变量,source ~/.bashrc

  • 测试,输入nvcc -V,输出如下内容

2、cuDNN安装

  • 前往nVidia官网下载cuDNN,下载时注意选择版本,这里以11.X为例

  • 下载到Windows系统上后,将文件复制到WSL的Ubuntu系统中,注意 ,在WSL中,Windows系统的路径变为/mnt/c/Users/zhangsan/Desktop/xxxxx

  • 解压文件,将include、lib文件夹下的文件复制到/usr/local/cuda-11文件夹下

    bash 复制代码
    # 复制     
    sudo cp -r lib/* /usr/local/cuda-11/lib64/
    sudo cp -r include/* /usr/local/cuda-11/include/
    
    # 修改权限
    sudo chmod a+r /usr/local/cuda-11/include/cudnn*
    sudo chmod a+r /usr/local/cuda-11/lib64/libcudnn*
  • 测试,输入 cat /usr/local/cuda-11.8/include/cudnn_version.h | grep CUDNN_MAJOR -A 2,结果如下:

3、Anaconda安装及清华镜像配置

4、TensorFlow-GPU安装

  • 经过测试,python版本指定为3.9,使用conda install tensorflow-gpu,即可自动安装,具体可参考CSDN文章【Tensorflow2.x】tensorflow-gpu 在 Ubuntu 上的安装

  • 测试是否可以调用GPU: tf.test.is_gpu_available(),输出True,即证明安装成功

  • 注意:建议不要使用tensorflow官方提供的pip命令进行安装(暂未使用该方法安装成功)

  • 代码中输入以下内容,可简单启用GPU

    python 复制代码
    import os
    os.environ["CUDA_VISIBLE_DEVICES"] = "0"

5、PyTorch-GPU安装

  • 前往PyTorch官网,选择需要的环境,复制conda命令,执行

  • 测试GPU是否可用:torch.cuda.is_available(),输出True,即证明安装成功

6、VSCode连接WSL

  • 在插件商店中下载WSL插件,安装后点击左下角即可

参考资料:

相关推荐
老艾的AI世界6 小时前
AI翻唱神器,一键用你喜欢的歌手翻唱他人的曲目(附下载链接)
人工智能·深度学习·神经网络·机器学习·ai·ai翻唱·ai唱歌·ai歌曲
sp_fyf_20249 小时前
【大语言模型】ACL2024论文-19 SportsMetrics: 融合文本和数值数据以理解大型语言模型中的信息融合
人工智能·深度学习·神经网络·机器学习·语言模型·自然语言处理
CoderIsArt9 小时前
基于 BP 神经网络整定的 PID 控制
人工智能·深度学习·神经网络
z千鑫10 小时前
【人工智能】PyTorch、TensorFlow 和 Keras 全面解析与对比:深度学习框架的终极指南
人工智能·pytorch·深度学习·aigc·tensorflow·keras·codemoss
EterNity_TiMe_10 小时前
【论文复现】神经网络的公式推导与代码实现
人工智能·python·深度学习·神经网络·数据分析·特征分析
思通数科多模态大模型11 小时前
10大核心应用场景,解锁AI检测系统的智能安全之道
人工智能·深度学习·安全·目标检测·计算机视觉·自然语言处理·数据挖掘
数据岛11 小时前
数据集论文:面向深度学习的土地利用场景分类与变化检测
人工智能·深度学习
学不会lostfound11 小时前
三、计算机视觉_05MTCNN人脸检测
pytorch·深度学习·计算机视觉·mtcnn·p-net·r-net·o-net
红色的山茶花11 小时前
YOLOv8-ultralytics-8.2.103部分代码阅读笔记-block.py
笔记·深度学习·yolo
白光白光11 小时前
凸函数与深度学习调参
人工智能·深度学习