在平时工作用的图形工作站上搭建一个GPU版本的Tensorflow的运行环境,看到网上说搭建过程中遇到很多问题,很多安装后无法进入登陆界面,或者在登陆界面循环等等,记录下自己的安装过程,备忘。
安装CUDA 8.0
由于我的运行环境是Ubuntu 14.04,只能安装CUDA 8.0。
deb安装失败的,执行清理
sudo apt-get --purge remove nvidia*
run安装失败的,执行清理
$ sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl$ sudo /usr/bin/nvidia-uninstall
仍有问题的,敲下命令
$ sudo apt-get autoremove --purge nvidia-* #把nvidia驱动清个干干净净$ sudo reboot
验证你的GPU是否支持CUDA
lspci | grep -i nvidia
根据显示的型号到官网查看https://link.jianshu.com/?t=https://developer.nvidia.com/cuda-gpus
下载CUDA安装包
从这里https://developer.nvidia.com/cuda-toolkit-archive选择对应的版本。
Disabling Nouveau
$ lsmod | grep nouveau
如果有内容输出,则需禁掉nouveau
$ sudo vi /etc/modprobe.d/blacklist-nouveau.conf
添加如下内容:
blacklist nouveauoptions nouveau modeset=0
保存退出(:wq),执行
$ sudo update-initramfs –u
然后重启电脑
$ sudo reboot
再执行
$ lsmod | grep nouveau
若无内容输出,则禁用成功
Ctrl+Alt+F1进入命令行模式
关闭图形化界面
$ sudo service lightdm stop
切换到cuda_8.0.54_linux.run的目录,执行
$ sudo chmod +x cuda_8.0.54_linux.run$ sudo ./cuda_8.0.54_linux.run
安装的时候,要让你先看一堆文字,我们直接不停的按空格键直到100%;
遇到提示是否安装openGL ,选择no,其他的可以一路accept, yes或回车。
重启图形化界面
$ sudo service lightdm start
这一步执行后我的电脑卡死了,强制断电源重启,卡在启动界面。ctrl+alt+F1也无法进入命令行界面,强制重启,又可以正常进入Ubuntu,不知道什么原因。明天继续安装。
閱讀更多 半杯茶的小酒杯 的文章