ubuntu16.04安装nivida显卡驱动以及cuda8.0，cudnnv6

每到一个地方就需要自己配置深度学习环境，整理下配置过程仅供大家参考。

安装双系统ubuntu或者直接装ubuntu，这比较简单没有什么坑可以讲的，直接从开机开始。

1.设置软件源中科大服务器（阿里云，163也可以）
2.sudo apt-get update
sudo apt-get upgrade
sudo apt-get dist-upgrade
sudo apt-get install build-essential

a) 检查自己的GPU是否是CUDA-capable

在终端中输入： $ lspci | grep -i nvidia ，会显示自己的NVIDIA GPU版本信息
去CUDA的官网查看自己的GPU版本是否在CUDA的支持列表中

b) 检查自己的Linux版本是否支持 CUDA(Ubuntu 16.04没问题)

c) 检查自己的系统中是否装了gcc

在终端中输入： $gcc –version 可以查看自己的gcc版本信息

d) 检查是否安装了kernel header和 package development

在终端中输入： $uname –r 可以查看自己的kernel版本信息
在终端中输入：$ sudo apt-get install linux-headers-$(uname -r)
可以安装对应kernel版本的kernel header和package development

a) 禁用 nouveau

终端中运行：$ lsmod | grep nouveau，如果有输出则代表nouveau正在加载。

Ubuntu的nouveau禁用方法：
在/etc/modprobe.d中创建文件blacklist-nouveau.conf，在文件中输入一下内容

blacklist nouveau
options nouveau modeset=0

打开tty终端，运行（ctrl+alt+f1）退出（ctrl+alt+f7）

$ sudo update-initramfs –u

设置完毕可以再次运行 $ lsmod | grep nouveau 检查是否禁用成功，如果运行后没有任何输出，则代表禁用成功。
============
下载对应版本显卡驱动，关闭主板快速启动和安全启动，若未关闭则安装过程中会报错

============
sudo service lightdm stop

卸载之前nivdia驱动：sudo apt-get autoremove --purge nvidia*

卸载刚装的驱动：sudo sh NVIDIA-Linux-x86_64-430.14.run -uninstall

安装驱动

sudo sh NVIDIA-Linux-x86_64-430.14.run -no-opengl-files -no-x-check -no-nouveau-check(不会进去死循环)

-no-x-check 对应sudo service lightdm stop 如果有这句就不用加了

-no-nouveau-check 对应blacklist-nouveau.conf 文件，如果有也不用加了

sudo service lightdm start

驱动安装完成
打开终端进去cuda文件下

sudo sh cuda.run

ubuntu16.04安装nivida显卡驱动以及cuda8.0，cudnnv6

因为已经装过显卡驱动了所以 Install NVIDIA Accelerate Graphics 为n

CUDA 8.0 Samples? 就是一些例子可装可不装，选择了n

配置cuda环境变量

sudo vim /etc/profile
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
source /etc/profile

echo $PATH
echo $LD_LIBRARY_PATH查看环境变量是否生效

sudo pip install tensorflow-gpu==1.4.0 指定tensorflow版本，该版本需要cuda8，cudnnv6

配置cudnn

解压缩：会生成cuda/include和cuda/lib64

将cuda/include目录中的cudnn.h文件拷贝到/usr/local/cuda-8.0/include/目录下

将cuda/lib64目录中的库拷贝到/usr/local/cuda-8.0/lib64/目录下；

将/usr/local/cuda-8.0/lib64/目录下的libcudnn库的权限改为777，执行：

sudo chmod 777 libcudnn*
ubuntu16.04安装nivida显卡驱动以及cuda8.0，cudnnv6 至此，环境配置完毕。

source /etc/profile

sudo ldconfig /usr/local/cuda-8.0/lib64 该命令解决pycharm不能导入cuda包

卸载cuda

sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl

sudo rm -rf /usr/local/cuda-8.0/

https://www.cnblogs.com/pprp/p/9540523.html这个安装文章比较好，值得推荐

ubuntu16.04安装nivida显卡驱动以及cuda8.0，cudnnv6

相关推荐