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,不知道什麼原因。明天繼續安裝。


分享到:


相關文章: