在平時工作用的圖形工作站上搭建一個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,不知道什麼原因。明天繼續安裝。
閱讀更多 半杯茶的小酒杯 的文章