1. 程式人生 > >Ubuntu16.04 cuda8.0+cudnn6.0+python3.5 配置tensorflow-gpu 1.4.0

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/