1. 程式人生 > >ubuntu14.04_64位安裝tensorflow-gpu

ubuntu14.04_64位安裝tensorflow-gpu

family 電腦配置 str 驅動 water list 出現 切換 名單

電腦配置:GeForce GTX 1080

安裝GTX1080驅動

進入英偉達官網,下載GTX1080驅動,開始搜索,然後下載所需版本即可。我下載的是最新的384.130。

技術分享圖片

也可以在這裏下載。

下載完成後,保存,作為備用用於刷新新驅動。

添加NVIDIA源

    sudo add-apt-repository ppa:graphics-drivers/ppa  

出現信息不用考慮,直接按回車即可。

sudo apt-get update   
sudo apt-get install nvidia-384  
sudo apt-get install mesa-common-dev   
sudo apt-get install freeglut3-dev

禁用nouveau

按Ctrl+Alt+F1進入命令提示符,新建一個黑名單文件:

sudo vi /etc/modprobe.d/blacklist-nouveau.conf  

輸入

blacklist nouveau    
options nouveau modset=0  

保存,退出。然後,執行

sudo update-initramfs -u    

檢查

lspci | grep nouveau 

如果沒有輸出,就代表正常。

安裝顯卡驅動

關閉桌面程序,如果不關閉,將無法完成安裝:

sudo service lightdm stop  #sudo start lightdm:重啟

ctrl+alt+F1:切換到命令行進行操作

#現切換到下載好的驅動包目錄
sudo sh NVIDIA-Linux-x86_64-384.130.run

然後重啟即可。

驗證安裝是否成功輸入

nvidia-smi

安裝cuda8.0

安裝cudnn 7

安裝java 8

ubuntu14.04_64位安裝tensorflow-gpu