一、安装显卡驱动
Ubuntu16.04.3在系统设置里可以直接安装NVIDIA驱动。
System Settings ->Software & Updates->Additonal Drivers->NVIDIA Corporation->Using NVIDIA binary driver ……
选择好之后点击Apply Changes,如果提示需要Secure boot,则按提示设置密码,之后重启时注意选择Secure boot, 并按提示输入密码。重启完成后,驱动便安装完毕。
二 、安装cuda8.0
虽然cuda9.0已经发布,但很多类库还不支持,所以推荐安装cuda8.0。
在NVIDIA官网下载cuda8.0,注意下载run文件格式。地址:https://developer.nvidia.com/cuda-80-ga2-download-archive
cuda需要安装OpenGL
sudo apt-get install build-essential libgl1-mesa-dev
sudo apt-get install freeglut3-dev
sudo apt-get install libglew-dev libsdl2-dev libsdl2-image-dev libglm-dev libfreetype6-dev
终端输入上述命令即可。
进入下载好的.run文件的目录,执行
sudo sh cuda_8.0.44_linux.run
安装cuda时注意不要安装驱动,其他选择yes即可。
配置环境变量:
sudo vim ~/.bashrc
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
:wq
source ~/.bashrc
创建链接文件:
sudo gedit /etc/ld.so.conf.d/cuda.conf
在打开的文件中添加如下语句:
/usr/local/cuda/lib64
然后执行
sudo ldconfig
使链接立即生效。
三、使用cudnn
下载cudnn,需要注册NVIDIA账号,地址https://developer.nvidia.com/rdp/cudnn-download
下载完cudnn5.0之后进行解压,cd进入cudnn5.1解压之后的include目录,在命令行进行如下操作:sudo cp cudnn.h /usr/local/cuda/include/ #复制头文件
再将cd进入lib64目录下的动态文件进行复制和链接:
sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5 #删除原有动态文件
sudo ln -s libcudnn.so.5.0.5 libcudnn.so.5 #生成软衔接
sudo ln -s libcudnn.so.5 libcudnn.so #生成软链接