Ubuntu 14.04搭建GPU版本的Tensorflow环境(1)-CUDA安装

Ubuntu 14.04搭建GPU版本的Tensorflow环境(1)-CUDA安装

在平时工作用的图形工作站上搭建一个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,不知道什么原因。明天继续安装。


分享到:


相關文章: