1.安装显卡驱动
首先卸载旧的显卡驱动
sudo apt-get purge nvidia*
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
添加官方ppa源并且刷新软件库
ubuntu-drivers devices
检测你的NVIDIA图形卡和推荐的驱动程序的模型
会显示有推荐安装程序,进行安装
sudo apt-get hold nvidia-driver-xxx
重启系统,检测
nvidia-smi
将会看到显卡信息
2. 安装CUDA
根据文档(https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html) 查看自己驱动对应的CUDA工具包版本,然后到网页(https://developer.nvidia.com/cuda-toolkit-archive) 下载自己的对应版本,本次选择的是deb(local)
使用给的命令进行安装,减少设置,最后使用
nvcc -V
检查安装情况
3. 下载cuDNN
打开网页(https://developer.nvidia.com/zh-cn/cudnn) 下载cuDNN需要账户登录,安装对应的CUDA版本下载对用的,下载之后进行解压
tar -zxvf cudnn-XXXXXX
会解压到cuda
文件夹中,然后按照对应子文件夹的归属,将文件复制到安装的CUDA驱动文件夹即/usr/local/cuda/
中
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
我在复制的时候直接复制了全部,直接使用了*
,最后检查清理使用
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
但是我并没有出现这个显示,但是深度学习程序还是直接测试成功了