目录
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
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
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已正确安装,如下图所示。
至此,安装完成。