Ubuntu16.04 cuda8.0+cudnn6.0+python3.5 配置tensorflow-gpu 1.4.0
機子情況
win10+ubuntu16.04雙系統
cpu:intel i7-7700
gpu:GTXI060 6G EVGA
在ubuntu16.04上想配置GPU版的tensorflow
備註:
我原來系統是win10,後來用u盤製作啟動盤安裝了ubuntu16.04 LTS
安裝驅動
先去官網下自己顯示卡型號對應的驅動,不贅述
解除安裝掉原有驅動
sudo apt-get purge nvidia*
禁用nouveau
我不太清楚雙系統是不是有什麼特殊性,我試了幾個常見的方法都不行。最後是以下這個方法奏效了
參考連結:https://blog.csdn.net/u014797226/article/details/79626693
1、開啟編輯配置檔案:
sudo gedit /etc/modprobe.d/blacklist.conf
2、在最後一行新增:
blacklist nouveau
3、改好後執行命令:
sudo update-initramfs -u
4、重啟使之生效:
sudo reboot
5、檢查是否成功
lsmod | grep nouveau
沒有任何輸出即成功
開始安裝驅動
我按照上面那個參考連結的建議做法試了,會報錯,說什麼x服務仍在執行,還有一個錯誤不懂說什麼。
然後我自己是這樣試成功的:
給予安裝檔案許可權
sudo chmod a+ x NVIDIA-Linux-x86_64-390.59.run (下載的驅動檔名)
然後
sudo ./NVIDIA-Linux-x86_64-390.25.run --no-opengl-files --no-x-check --no-nouveau-check
-–no-opengl-files
只安裝驅動檔案,不安裝OpenGL檔案。這個引數最重要
--no-x-check
安裝驅動時不檢查X服務
--no-nouveau-check
安裝驅動時不檢查nouveau
參考連結裡頭說,後面兩個引數可不加。但我試過後發現,全部加上才行
注意no那裡是雙槓
重啟,沒有問題,輸入命令:nvidia-smi
顯示出顯示卡對應資訊,則安裝成功
1.新增conf 檔案:cat /etc/modprobe.d/blacklist-nouveau.confblacklist nouveauoptions nouveau modeset=0
2. 重新生成 kernel initramfs:sudo update-initramfs -u
3. 3. 重啟:sudo reboot
參考連結:
https://www.zhihu.com/question/68565667/answer/264871312
http://www.cnblogs.com/Qwells/p/6086773.html#undefined
https://blog.csdn.net/zafir_410/article/details/73188228?utm_source=itdadao&utm_medium=referral
安裝CUDA8.0
注意一點:
在安裝cuda自帶的驅動時,一定要選N,否則之前的就白裝了
安裝CUDNN6.0
檢視是否安裝好
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
安裝Anaconda
https://blog.csdn.net/u012318074/article/details/77074665
官網下載
使用cd
命令切換到下載目錄
使用ls
命令列出檔案目錄
bash Ana
按tab鍵,回車
進入tensorflow環境
啟用
conda activate tensorflow-1.4.0
退出
conda deactivate
測試安裝是否成功
在Pycharm設定tensorflow的環境變數
terminal裡面是可以用的,但是不太方便,安裝了pycharm後,發現每次新建一個使用tensorflow的py檔案都要在debug設定那裡加環境變數:
export LD_LIBRARY_PATH=/usr/local/cuda/lib64/
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64/