前言
雖然有很多教程都給出了如何在Ubuntu上安裝Tensorflow,但是卻沒有一篇文章講如何在Deepin系統(tǒng)安裝Tensorflow,這里將安裝過程的幾個(gè)要點(diǎn)講解一下
流程
- 首先下載Nvidia的Linux驅(qū)動(dòng)。在官網(wǎng)可以下載。然后運(yùn)行,如果不能運(yùn)行,則需要修改驅(qū)動(dòng)運(yùn)行權(quán)限,使用chmod u+x來添加運(yùn)行權(quán)限。關(guān)于deepin安裝顯卡驅(qū)動(dòng),可以參考這個(gè)文章 https://wiki.deepin.org/index.php?title=%E6%98%BE%E5%8D%A1
- 安裝Docker-ce。這里一定要遵循官方的安裝方案,只此一個(gè)方法可以正常安裝最新版docker。https://wiki.deepin.org/index.php?title=Docker
- 下載Nvidia-docker。在Github的Nvidia-Docker頁面找到Ubuntu的安裝包和安裝步驟,安裝即可
- 接下來運(yùn)行
docker run -it -p 8888:8888 tensorflow/tensorflow:latest-gpu此時(shí)可以進(jìn)入官方的tensorflow帶GPU版本的鏡像啦,打開localhost:8888可以看到j(luò)upyter的頁面
坑
在安裝Nvidia顯卡驅(qū)動(dòng)的時(shí)候,根據(jù)安裝軟件的提示,需要關(guān)閉Linux的桌面系統(tǒng)X server,此時(shí)可以通過tty1~6來進(jìn)行終端操作。但是如果Deepin已經(jīng)安裝了自帶的Nvidia驅(qū)動(dòng),要停止lightdm服務(wù)時(shí),在Deepin在關(guān)閉X server的同時(shí),顯示器也關(guān)閉了,無法顯示tty1 ~6的屏幕,由于這個(gè)問題,導(dǎo)致Nvidia驅(qū)動(dòng)無法安裝。即使手動(dòng)卸載安裝的Deepin官方Nvidia驅(qū)動(dòng),經(jīng)過了繁瑣的卸載指令,在切換到tty1 ~6后,在關(guān)閉圖形界面的時(shí)候,仍然會(huì)導(dǎo)致tty屏幕被關(guān)閉,黑屏一片,無法繼續(xù)后面的驅(qū)動(dòng)卸載和重新安裝官方Nvidia驅(qū)動(dòng)。
解決方案
為了避免Nvidia驅(qū)動(dòng)無法安裝,又能簡(jiǎn)潔科學(xué)的操作,所以必須在第一次安裝Deepin系統(tǒng)的時(shí)候,在未安裝顯卡驅(qū)動(dòng)的情況下對(duì)電腦進(jìn)行操作進(jìn)行安裝Nvidia顯卡官方linux驅(qū)動(dòng),首先按下快捷鍵“Ctrl+Alt+F2”,進(jìn)入tty2,然后輸入sudo systemctl stop lightdm 停止lightdm服務(wù),此時(shí)電腦關(guān)閉X server的時(shí)候不會(huì)導(dǎo)致顯示器的關(guān)閉,之后運(yùn)行chmod u+x NVIDIA-Linux-x86_64-352.55.run #賦予可執(zhí)行權(quán)限 以及sudo ./NVIDIA-Linux-x86_64-352.55.run #安裝驅(qū)動(dòng)文件 然后重啟就可以正常的使用官方閉源驅(qū)動(dòng)啦。
后續(xù)
使用GPU來運(yùn)行Tensorflow程序是非??斓?。一般的程序可以提速10倍以上,有些程序可以提速50到100倍的速度,所以使用GPU來進(jìn)行Tensorflow編程是非常有必要的。
另外,不知道是Deepin的問題還是Nvidia-Dcoker的問題還是驅(qū)動(dòng)的問題,在電腦待機(jī)之后會(huì)導(dǎo)致Tensorflow鏡像出錯(cuò),所以在訓(xùn)練的時(shí)候不要讓機(jī)器進(jìn)行待機(jī)動(dòng)作。