GPU驱动,CUDA以及cuDNN一直在持续更新中,现在的安装也越来越简单!
1. 现在的CUDA安装包中自带的GPU驱动已经满足所有需求,不需要提前安装GPU驱动;所以直接从下边网址下载
https://developer.nvidia.com/cuda-downloads
Linux--x86_64--Ubuntu--16.04/18.04--runfile(local)
2. Ctrl+Alt+F1进入命令行模式,依次执行
sudo service lightdm stop
chomod +x 下载的run文件
sudo ./下载的run文件
sudo service lightdm start
安装过程中,所有部件都安装(默认全部安装),安装完之后,打开./bashrc文件:gedit ~/.bashrc,在末尾加上
export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH
然后保存文件,执行source命令: source ~/.bashrc
然后运行命令:nvidia-smi 检查GPU安装情况,nvcc -V 检查CUDA安装情况,会显示GPU和CUDA的版本。
3.安装cuDNN
打开网址https://developer.nvidia.com/rdp/cudnn-download
下载三个文件
cuDNN Runtime Library for Ubuntu16.04 (Deb)
cuDNN Developer Library for Ubuntu16.04 (Deb)
cuDNN Code Samples and User Guide for Ubuntu16.04 (Deb)
打开terminal,用dpkg -i 命令依次安装上述三个文件,一个例子如下:
sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-devel_7.0.3.11-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb
4.检查cuDNN安装情况
cp -r /usr/src/cudnn_samples_v7/ $HOME
cd $HOME/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN
最终看到屏幕输出 Test passed! 表示一切安装顺利!