深度学习环境搭建（3）Deepin20.3+CUDA10.2+CUDNN8.3+PyTorch1.8

安装CUDA10.2：

 

下载地址：CUDA Toolkit Archive | NVIDIA Developer

这里我选择的是cuda10.2

 
https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.runhttps://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run

可以直接将链接输入至迅雷，可以提速。

sudo sh cuda_10.2.89_440.33.01_linux.run

我使用官方给的命令下载会报错，需要在命令后面加一些东西 

sudo sh cuda_10.2.89_440.33.01_linux.run  --silent --toolkit --samples --librarypath=/usr/local/cuda-10.2

 使用官方给的命令时会出现以下页面

取消驱动！！！

 因为Deepin20.1取消了dedit的命令，这里我将deepin-editor命令修改为dedit。

sudo ln -sf /bin/deepin-editor /bin/dedit

 添加环境变量

sudo dedit ~/.bashrc

ubuntu系统则需使用sudo gedit ~/.bashrc

在文件打开的末尾添加：

export PATH="/usr/local/cuda-10.2/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH"

保存并退出

使其生效：

source ~/.bashrc

查看CUDA是否成功安装：

 nvcc -V

 出现以上界面说明CUDA安装成功。

安装CUDNN8. 3：

下载地址：cuDNN Archive | NVIDIA Developer

选择cuDNN Library for Linux,笔者下载的是8.3版本，老版本的安装方法会有些不同。

 对其解压缩：

tar -xvf cudnn-10.2-linux-x64-v8.3.0.98.tgz

 执行以下命令：

sudo cp cuda/include/cudnn* /usr/local/cuda-10.2/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64/
sudo chmod a+r /usr/local/cuda-10.2/include/cudnn*
sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*

验证cuDNN是否安装成功：

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

 出现cuDNN版本号说明安装成功

 安装pytorch：

pip换源：

没有换源的可以参考一下

创建文件夹：

 mkdir ~/.pip

建立一个新文件 

cd .pip
touch pip.conf
chmod +w pip.conf 
dedit pip.conf      //ubuntu用gedit 

写入以下内容

[global]
timeout = 60000
trusted-host = pypi.douban.com
index-url = http://pypi.douban.com/simple

 查看pip的版本：

pip3 -V

 pip版本要在20以上，19需要更新

python3 -m pip install --user --upgrade pip

正式安装pytorch ：

进入pytorch官网查看与cuda对应的版本Previous PyTorch Versions | PyTorch 

pip3 install torch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0

 我直接输入以上命令会报错

于是我先安装的 typing-extensions

pip3 install typing-extensions

然后再次安装： 

pip3 install torch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0

安装成功：