ubuntu16.04+1080显卡+cuda8.0+cudnn5.1安装记录

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

显示以下信息即正确安装:

ubuntu16.04+1080显卡+cuda8.0+cudnn5.1安装记录_第1张图片

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

出现以下结果即是安装成功:

ubuntu16.04+1080显卡+cuda8.0+cudnn5.1安装记录_第2张图片

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安装教程

你可能感兴趣的:(ubuntu)