每到一个地方就需要自己配置深度学习环境,整理下配置过程仅供大家参考。
安装双系统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这个安装文章比较好,值得推荐