ubuntu 16.04 安装NVIDIA,cuDNN
cuda安装
官网下载cuda:https://developer.nvidia.com/cuda-downloads,我下载的为cuda-8.0版本
cuda安装步骤如下:
- sudo apt-get install mesa-common-dev freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
- /etc/modprobe.d/blacklist.conf添加:
- blacklist amd76x_edac
- blacklist vga16fb
- blacklist nouveau
- blacklist rivafb
- blacklist nvidiafb
- blacklist rivatv
- 更新内核: sudo update-initramfs -u
- 卸载旧驱动:sudo apt-get remove --purge nvidia*
- ctrl+alt+f1进入终端界面。 sudo service lightdm stop
- sudo sh cuda8.0.27linux.run --no-opengl-libs --verbose 进行安装一路enter和y即可。
- 配置环境变量 在/etc/profile添加:
- PATH=/usr/local/cuda-8.0/bin:$PATH
- export PATH
- 执行 source /etc/profile使其生效
- 添加库搜索路径。在 /etc/ld.so.conf.d/添加文件cuda.conf,内容为:
- /usr/local/cuda-8.0/lib64
- 执行 sudo ldconfig使其生效
- 执行nvidia-smi测试是否安装成功
cudnn安装:
官网下载cudnn:https://developer.nvidia.com/cudnn
安装cudnn步骤如下:
sudo cp lib* /usr/local/cuda/lib64/
sudo cp cudnn.h /usr/local/cuda/include/