Ubantu20.04 安装 CUDA,CUDNN,pytorch

1.安装显卡驱动

查看我的另一篇文章 

2.安装cuda

查看cuda版本   版本号为11.6

找到合适的CUDA版本

点进去一个，注意 箭头标记的地方，要和自己的驱动一致，不一致的就换另一个！

复制https:.....这一行，在 浏览器打开

找到下载的.run文件进行安装

sudo sh cuda_11.6.1_510.47.03_linux.run 

因为驱动已经安装过了，所以把第一个driver选项取消

安装完成： 

 安装完成后修改环境变量

sudo gedit ~/.bashrc

export PATH=$PATH:/usr/local/cuda/bin  
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64  
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64

 修改后使其生效

source ~/.bashrc

测试，成功

3.安装cudnn

cuDNN Archive

找到适合自己cuda版本的 

两个任选一个，找到合适自己系统的版本

下载 cudnn-linux-x86_64-8.4.0.27_cuda11.6-archive.tar.xz 并提取，修改目录文件名为cudnn。

将提取的文件 复制到指定的文件夹

sudo cp cudnn/include/cudnn.h /usr/local/cuda/include/
sudo cp cudnn/include/cudnn_version.h /usr/local/cuda/include/
sudo cp cudnn/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*

查看是否安装成功

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

4.安装Anaconda

见我的另一篇文章

5.安装pytorch

Start Locally | PyTorch

 选择合适的pytorch版本

conda install pytorch torchvision torchaudio cudatoolkit=11.6

复制命令 在终端输入   去掉 -c pytorch，等待安装成功

很遗憾，安装不成功。

使用另一位博主的方法：

CUDA11.6安装PyTorch三件套

链接如下：

1. 下载PyTorch

pip install torch --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116

2. 下载Torchvision

pip install torchvision --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116

3. 下载Torchaudio

pip install torchaudio --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116

我发现下载的cudnn版本也是11.3,可以选择博主的方法，也可以上一步中选择合适的pytorch版本的时候选择11.3。

进入需要安装torch 的环境中，执行推荐的命令

conda install pytorch torchvision torchaudio cudatoolkit=11.3

​查看是否可用