1. 显卡驱动安装
(1)在nvidia官网上下载相应的驱动:NVIDIA-Linux-x86_64-384.111.run
(2)卸载可能存在的旧版本 nvidia 驱动(对没有安装过 nvidia 驱动的主机,可省略)
在终端输入:
sudo apt-get remove --purge nvidia*
(3)安装驱动可能需要的依赖
sudo apt-get update
sudo apt-get install dkms build-essential linux-headers-generic
(4)把 nouveau 驱动加入黑名单
sudo nano /etc/modprobe.d/blacklist-nouveau.conf
在文件 blacklist-nouveau.conf 中加入如下内容:
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off
(5)禁用 nouveau 内核模块
echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf
sudo update-initramfs -u
(6)重启
(7)重启后再次进入字符终端界面,并关闭图形界面(ALT+CTRL+F1)
sudo service lightdm stop
(8)安装驱动
sudo chmod u+x NVIDIA-Linux-x86_64-384.111.run
sudo ./NVIDIA-Linux-x86_64-384.111.run
(9)重启
(10)重启后,在终端输入:
nvidia-smi
显示以下信息即正确安装:
2. cuda8.0安装
从NVIDIA下载cuda安装文件:cuda_8.0.61_375.26.linux.run
(1)
sudo sh cuda_8.0.61_375.26_linux.run
会出现一串须知一样的信息,可以直接ctrl+c跳过,也可按空格阅读
阅读完信息后,会有一些选择,需要注意的是,是否安装NVIDIA驱动问题时,应选择否。其他选择yes或者默认即可。
安装成功后会出现如下界面:
===========
= Summary =
===========
Driver: Not Selected
Toolkit: Installed in /usr/local/cuda-8.0
........
***WARNING: Incomplete installation! Tis installation did not install the CUDA Driver. A driver ......
.....
Logfile is /opt/temp//cuda_install_6583.log
安装完毕后,需声明环境变量,并将其写入到~/.bashrc的尾部:
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda
(2)测试cuda
cd ~/NVIDIA_cuda-8.0_Samples/1_Utilities/deviceQuery
make
./deviceQuery
出现以下结果即是安装成功:
3. 安装cuDNN
下载cuDNN文件:cudnn-8.0-linux-v5.1.tgz并解压
(1)复制头文件
cd ~/cuda/include
sudo cp cudnn.h /usr/local/cuda/include/
(2)lib64目录下动态文件进行复制和连接
cd ~/cuda/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 #生成软链接
参考:ubuntu16.04安装NVIDIA显卡驱动
Ubuntu16.04下安装CUDA8.0和tensorflow
Ubuntu16.40+cuda8.0+caffe安装教程