Ubuntu18.04安装cuda10.1+cudnn8.0.5+pytorch1.8.1【亲测～】
目录
第一步 ：Cuda10.1的安装
1）输入命令，确保自己没装Cuda：
cat /usr/local/cuda/version.txt
在命令nvidia-smi得到的CUDA Version：11.5等版本
，这里的意思是你的显卡驱动支持的最大CUDA版本，并不是真正安装的CUDA版本（所以麻烦不要误人子弟）
具体来说：
CUDA Driver = CUDART,
CUDA Driver Version = 11.5,
CUDA Runtime Version= 10.1
CUDA驱动版本和CUDA运行版本，这两者是不一样的东西～
2）进入Cuda官网，当然你也可以直接百度cuda10.1，选择CUDA Toolkit 10.1 update1 Archive | NVIDIA Developer
这里有很多不同的安装类型(Installer Type)，建议选用第一个.run安装包，不要问为什么，因为干净简洁～
下载后得到：
如右下角所示：2.4G，然后执行：
sudo sh cuda_10.1.168_418.67_linux.run
敲击accept，下一步：
这里如果直接选择Install的话，报错：
Installation failed. See log at /var/log/cuda-installer.log for details
因为你已经有了CUDA驱动，所以选择第一行按回车，回车键作用是将 [X] 就会变成[ ]，[X]代表有，[ ]代表无。
PS：图我没保存，就借用了别人的，不要计较版本不对应，我还是Cuda10.1~
最后选择yes，直到结束。
3）确认cuda是否安装成功
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
再执行：
sudo make
最后执行：
./deviceQuery
看见Result = PASS意味着成功~
4）添加环境变量
sudo gedit ~/.bashrc
打开该文件，然后再最后输入：
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1
注意：当中的cuda版本要对应你下载的版本
最后，使之生效：
source ~/.bashrc
第二步 ：Cudnn8.05的安装
1.进入官网：https://developer.nvidia.com/rdp/cudnn-archive下载
或者你也可以直接百度cudnn进入
然后，需要登录，选择Sign in（如已有帐号自动忽略）
注册后，打开页面：
选择ubuntu中的三个文件，都需要下载：
完成后，cd到下载的文件夹，执行：
sudo dpkg -i libcudnn8_8.0.5.39-1+cuda10.1_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.5.39-1+cuda10.1_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.5.39-1+cuda10.1_amd64.deb
2.确认cudnn安装成功
通过上步骤的安装，得到了cudnn的安装路径：/usr/include/
因此，执行：
sudo cp /usr/include/cudnn.h /usr/local/cuda/include
sudo chmod a+x /usr/local/cuda/include/cudnn.h
复制cudnn的头文件到cuda文件夹中，然后确认cudnn安装是否成功：
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cp -r /usr/src/cudnn_samples_v8/ $HOME
cd $HOME/cudnn_samples_v8/mnistCUDNN
make clean && make
./mnistCUDNN
输出Test passed，表示成功～
第三步 ：pytorch的安装
1）推荐使用pip3安装，没有pip3的执行：
sudo apt install python3-pip
2）进入pytorch官网，
找到cuda 10.1
执行：
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
安装完成～
测试：
python
import torch
torch._version__
3）一般的，Ubuntu会自带两种python，默认的是python2.7
输入python确认：
输入python -v和python3 -v显示软连接：
看到有python 3.6，然后也可以cd /usr/bin/利用ls命令查看python的版本，并删除python文件夹建立新的软连接，我不敢～
于是可以：
sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 150
sudo update-alternatives --config python
转换一下即可，再输入python显示：Python 3.6.9 (default, Mar 15 2022, 13:55:28)
PS:修改ubuntu光标的粗细，因为我的光标总是莫名其妙的消失，所以我想换一下：
右键选择：preferences（配置）
选择cursor shape光标形状：
