一文搞定Ubuntu22.04（双系统）配置pytorch环境

目录

1.安装nvidia驱动

2.安装CUDA

 3.安装cudnn

4.安装anaconda

5.创建虚拟环境

6.安装pytorch

---

1.安装nvidia驱动

解决nvidia-smi报错

如图则安装成功

2.安装CUDA

查看驱动与cuda的对应

查看pytorch和cuda版本对应

安装链接

配置cuda环境

考虑到pytorch和cuda版本的对应，笔者选择了cuda11.7

最后效果

 3.安装cudnn

cudnn

选择合适的版本，笔者选了v8.9.0 for 11.x linux

解压命令

tar zxvf cudnn-11.2-linux-x64-v8.1.1.33.tgz -C .

也可以打开文件夹，解压到此处

 并进入到相应目录，执行以下命令

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

以上配置cuDNN成功

验证是否成功

cat /usr/local/cuda-11.1/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

 cudnn版本8.9.0

参考：测试cudnn是否安装成功

4.安装anaconda

清华anaconda镜像网站

安装步骤

5.创建虚拟环境

pytorch和torch的版本对应

torch和python的版本对应

通过对比，11.7的pytorch接受的最低python为3.8，遂选择python3.8创建虚拟环境

创建conda虚拟环境

 conda create -n pytorch python=3.8

 创建成功并进入虚拟环境

6.安装pytorch

pytorch官网

 根据cuda版本选择pytorch版本，在虚拟环境下输入

pip3 install torch torchvision torchaudio

检验是否安装成功

 文章到此结束，小白记录一下安装过程，如有错误请批评指正