Tensorflow-gpu Ubuntu16.04 GTX1060 雙顯卡

Tensorflow1.4.0不支持cuda9.0

一開始在cuda官網上下載了cuda9.0,非常不幸,一番辛苦下來發現現在最新版本的tensorflow-gpu不支持cuda9.0。因此,只能在網上尋找cuda8.0進行下載

卸載已有的cuda

嘗試了網上很多已有的命令,這個命令可以使用

sudo apt-get autoremove --purge cuda

如果沒有卸載乾淨,可以使用

rm -rf <文件夾路徑>

將剩餘的整個文件夾刪除(包括文件夾下的所有文件)

安裝Anaconda2

anaconda2是一個非常好的集成環境,裏面包含了python,以及spyder IDE,PIP等工具。注意安裝好anaconda2後需要配置環境變量。

pip 安裝 tensorflow

使用

pip install tensorflow-gpu

安裝 tensorflow的gpu版本

下載cuda8.0

推薦尋找百度網盤

下載cudnn6 for cuda 8.0

注意,截至2017-12-1, 當前版本的tensorflow1.4.0只是支持cudnn6,不要下成cudnn7

安裝cuda8.0和cudnn6

參考博文
http://blog.csdn.net/zhaoyu106/article/details/52793183
親測沒有問題,但是使用cudnn6的話,gcc不需要降低版本。

安裝cuda8.0,使用.run文件

sudo sh cuda_8.0.44_linux.run

我是在已經安裝了nvidia驅動的情況下進行安裝的,因此,提示是否安裝nvidia driver的時候,我選擇了no。其他都是一路accept,yes,或者回車

安裝cudnn6.0

cudnn其實類似於cuda的擴展,下載下來的是一個壓縮文件,解壓之後進行下面三行命令即可。(解壓出的文件默認名是cuda)

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

Nvidia 顯卡驅動安裝

可以參考
http://blog.csdn.net/qq_33843297/article/details/78683569

另外有個命令暫時這裏保存一下

sudo service lightdm stop  #用於關閉圖形化界面
sudo service lightdm restart  #用於開啓圖形化界面

使用ctrl+Alt+F1可以從linux的圖形界面進入命令行界面
使用ctrl+Alt+F6可以從linux的命令行界面返回圖形界面

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章