Ubuntu20.04安装cuda cudnn pytorch pycharm记录

记录时间：2021年1月31日
版本：Ubuntu20.04、cuda11.0、cudnn对应的版本、pytorch对应的版本。我的电脑安装win10+Ubuntu20.04双系统，中途会重启进入windows系统进行一些下载。

0.安装NVIDIA驱动

Ubuntu20.04系统安装完成后，已有NVIDIA驱动，窗口输入nvidia-smi后显示驱动的信息。因此无需另外下载、安装驱动。此处显示驱动的版本，和最高支持的CUDA版本。“CUDA Version:11.2”指的是最高支持11.2版本的cuda，而后续不一定要安装11.2版本的cuda，也可以安装低于11.2的版本。

1.安装cuda

我们要配置pytorch的环境，首先要看pytorch支持哪些版本的cuda，而不能盲目选择。

（1）查看pytorch支持的cuda版本。

进入pytorch官网https://pytorch.org/，查看pytorch支持的cuda版本。

cuda 10.2针对的系统更偏向于Ubuntu18及以下的，我的系统是Ubuntu20.04，因此选择安装cuda11.0。

（2）下载cuda安装包并安装

进入cuda版本选择页面https://developer.nvidia.com/cuda-toolkit-archivE选择对应的版本；或直接再百度搜索cuda11进入指定版本的下载页面。

选择系统“Linux”，“x86_64”，“Ubuntu”，“20.04”。安装包的类型选择"runfile(local)"（因为其他类型的安装需要较多零散的安装包，runfile只有一个安装包，更便于下载）。wget会从官方网站下载，速度非常慢甚至会下载失败，因此我复制“wget”后面的网址，到windows系统中用迅雷下载此文件，大小上G。

[注：]如果迅雷下载的文件只有几个B大小，那肯定下载错了，需要重新复制链接进行下载。如果迅雷下载太慢甚至多次下载失败，则开会员可以解决。不同于去年我装系统用的更低版本的cuda，无需迅雷，直接官网下载速度也很快。要根据实际情况选择合适的下载方式。

在windows系统用迅雷下载好后，将.run安装包拷到Ubuntu系统中某个方便找到的文件夹下，在终端进入此文件夹，执行“sudo sh cuda_11.0.2_450.51.05_linux.run”，进入安装界面。安装选项除了450版本的driver不选（这是NVIDIA驱动，系统已经有了更高版本的，系统中NVIDIA驱动版本若是大于450.51.05，则不选；若小于此版本，则要选），其他都选。其他选项默认，很快就完成安装。

最后，添加环境变量。

sudo vim /etc/profile
#在最后添加两行
export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
#保存并退出
：wq 回车
#使它们生效
source /etc/profile 

在终端输入nvcc -V，若出现下图所示的结果，则说明安装成功、添加环境变量成功

验证cuda是否正常工作：依次在窗口中输入以下命令。输入最后一个命令，如果显示Result = Pass，则安装成功。

cuda-install-samples-11.0.sh ~  
cd ~/NVIDIA_CUDA-11.0_Samples 
make #需等待一段时间
./1_Utilities/deviceQuery/deviceQuery 

2.安装cudnn

进入cudnn官网https://developer.nvidia.com/cudnn登录后在Download的页面选择需要的版本。或直接进入版本选择页面https://developer.nvidia.com/cudnn。我选择下图的版本（为什么，因为其他版本我电脑上迅雷下载总是出错，这个不出错，所以选它）。

选择对应的安装包。我选择第一个（因为其他形式的安装包有好几个，就第一个安装包最简洁，所以选它）。同样，在windows系统中右键选择“迅雷下载”，下载此文件。

下载完成后，将文件拷到Ubuntu系统中。按照官方文档https://docs.nvidia.com/deeplearning/sdk/cudnn-install/的说明进行下载。我们下载了tar file，因此，参考官方文档的这一部分。

3.安装Anaconda（略）

这一部分网上有较多较好的教程，在此不赘述。建议从清华源等镜像源下载安装包。

4.conda换源、建立环境、pip换源

pytorch的安装会涉及到一些模块的下载，如果不换成国内的镜像源，往往导致下载失败。

（1）conda换源

在命令行输入命令 gedit ~/.condarc，去除-default，并添加以下内容：
[注：]我用这些源安装成功了，但还不完善，后续有更优质的源再继续添加。

auto_activate_base: false
show_channel_urls: true
channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.ustc.edu.cn/anaconda/pkgs/main/

（2）建立conda环境

后面所有的模块都安装在集成环境中，环境也将作为pycharm中的interpreter。建立名为pytorch，包含python3.6的环境

conda create -n pytorch python=3.6

激活环境

conda activate pytorch

关闭环境

conda deactivate

（3）pip换源

pip包含在建立的conda环境中，先进入名为pytorch的环境，然后将pip换源

conda activate pytorch
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

5.在环境中安装pytorch

进入名为pytorch的环境（环境的名字可以任意选取）

conda activate pytorch

回到pytorch的官网，查看安装的方法。去除末尾的 -c pytorch，只输入“conda install pytorch torchvision torchaudio cudatoolkit=11.0”，开始安装。

安装完成后，在pytorch环境的python中import torch，如果不报错，则安装成功.

6.安装pycharm

进入pycharm官网https://www.jetbrains.com/pycharm/download/#section=linux，选择社区版（不会过期，不用激活码）

安装包下载后，解压，在终端打开解压后的文件夹。

cd Downloads/pycharm-版本号/bin
sh ./pycharm.sh

如果弹出窗口，默认选择就行，这时已经可以进入pycharm软件界面。

cd /usr/share/applications
sudo gedit pycharm.desktop

#输入以下内容
[Desktop Entry]
Version=1.0
Type=Application
Name=Pycharm
Icon=/home/XXXXXXXX/Downloads/pycharm-版本号/bin/pycharm.png
Exec=sh /home/XXXXXXX/Downloads/pycharm-版本号/bin/pycharm.sh
MimeType=application/x-py;
Name[en_US]=pycharm

这时就能在软件列表里找到pycharm软件。右键“add to favorates”，就能把它添加到任务栏。

如果在任务栏仍不能启动pycharm，则先从刚刚下载的安装包中启动pycharm.sh。打开pycharm后，选择Tools----Create desktop entry，这时就能在软件中看到pycharm的图标，点击右键，“add to favorates”就能添加到桌面任务栏了。