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

 

因为已经装过显卡驱动了 所以 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*
至此,环境配置完毕。

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这个安装文章比较好,值得推荐

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值