Ubuntu20.04环境下Pytorch安装教程

一 安装CUDA

访问nvidia官网，下面的网址即可，找到自己对应的版本

CUDA Toolkit 11.5 Update 1 Downloads | NVIDIA Developer

下面为官网的命令行，如果觉得打开官网麻烦也可以直接使用下面的命令

wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run
sudo sh cuda_11.4.0_470.42.01_linux.run

 其实很简单，但是因为网络原因经常会遇见下载中断的情况，这个时候不要着急，直接重新运行上一次的代码就可以接着下载。

我在安装的时候也遇见了一个问题，就是来来两次下载99%之后都会报错，“段错误，核心已转储”。似乎是因为显存的问题，我还不是很确定，但是重启之后问题就消失了，等到以后我在来探究这个问题吧。

接下来就是漫长的等待————

终于下载完成了 ！然后就是安装

敲入accept，按下回车

如果Ubuntu环境下已经安装过驱动了的话，这里取消最上面的驱动，然后选择Install.

---

二  安装Cudnn

下载安装文件

按需求下载cudnn的安装文件：https://developer.nvidia.com/rd

解压下载下来的文件，然后在终端打开，输入如下命令

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

---

三 安装pytorch 

sudo pip3 install torch torchvision -i https://pypi.mirrors.ustc.edu.cn/simple/