Ubuntu安装CUDA、cuDNN、PyTorch

1.安装显卡驱动

1.1 Method 1

安装显卡驱动

进入System Settings——Software & Updates——Additional Drivers，选择Using NVIDIA binary driver-version xxx.yyy from nvidia-xxx(proprietary)，如下图所示。

验证显卡驱动

安装完成后，终端输入
nvidia-smi
显示显卡相关信息，如下图所示。

1.2 Method 2

禁用nouveau

修改配置文件，

sudo gedit /etc/modprobe.d/blacklist.conf

在配置文件末尾添加如下：

blacklist nouveau
options nouveau modeset=0

更新配置文件，

sudo update-initramfs -u

重启，检查是否禁用成功，

lsmod | grep nouveau

执行后如果没有输出，则已禁用成功。

卸载旧驱动

进入命令行界面，

Ctrl-Alt+F6

禁用lightdm服务，

sudo service lightdm stop

删除原有显卡驱动，

sudo apt-get remove --purge nvidia*
sudo chmod +x NVIDIA-Linux-x86_64-xxx.xx.run
sudo ./NVIDIA-Linux-x86_64-xxx.xx.run --uninstall

安装新驱动

安装驱动，

sudo ./NVIDIA-Linux-x86_64-xxx.xx.run

启用lightdm服务，

sudo service lightdm start

重启，安装完成。

reboot

参考Ubuntu安装和卸载CUDA和CUDNN

2.安装CUDA

2.1 下载及安装CUDA

根据显卡驱动版本选择CUDA版本，点此查阅。
点此下载CUDA，根据需求依次选择即可，如下图所示。
完成下载后，按提示完成安装CUDA。

2.2 添加PATH

终端执行以下指令打开.bashrc文件：

sudo gedit ~/.bashrc 

末尾添加以下两行，保存并更新.bashrc。

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

2.3 验证CUDA，查看CUDA版本

完成安装后，终端输入

cat /usr/local/cuda/version.txt
# 最新版本使用以下指令
nvcc -V

若CUDA已正确安装，如下图所示。

3.安装cuDNN

参考：NVIDIA cuDNN安装说明

3.1 下载cuDNN

点此下载cuDNN，根据CUDA版本选择cuDNN版本。
选择cuDNN Library for Linux即可，如下图所示。

3.2 安装cuDNN

按照此链接中 2.3.1. Installing From A Tar File 小节安装。

3.3 验证cuDNN，查看cuDNN版本

安装完成后，终端输入

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
# 最新版本使用以下指令
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

若cuDNN已正确安装，如下图所示。

4.安装PyTorch

4.1 安装PyTorch

参考：PyTorch官网
需要选择系统信息等，如下图所示。

若当前版本没有满足选项，可前往Previous PyTorch Versions，选择需要版本，如下图所示。

例如，本文安装指令为：
pip install torch torchvision

4.2 验证PyTorch

打开终端，进入Python，依次输入：

import torch
print(torch.version.cuda)

若PyTorch已正确安装，如下图所示。

至此，安装完成。