Hugh ubuntu 14.04搭建tensorflow-gpu開發環境
轉載地址:https://www.cnblogs.com/samo/p/8092708.html
一、安裝nvidia顯示卡驅動
去navidia官網檢視最新的驅動版本號:navidia官網:http://www.geforce.cn/drivers
找到顯示卡對應的驅動下載,例如下載的驅動為 NVIDIA-Linux-x86_64-375.82.run
ubuntu桌面環境下,按ctrl + alt +F1進入text mode 登入賬戶,輸入下面命令,一路ok
1 2 3 |
|
會重新進入圖形桌面
點選左邊的設定,點選詳細資訊,在圖形標籤欄是否有NVIDIA的標識,如有,則表示驅動安裝成功
二、安裝cuda-8.0
官方網站下載:CUDA Toolkit 8.0。下載網址為:https://developer.nvidia.com/cuda-80-ga2-download-archive
下載完後,執行下命令:
1 2 3 |
local -ga2_8.0.61-1_amd64.deb
|
1 2 3 |
|
這時在/usr/local目錄下產生一個cuda安裝的路徑叫"cuda-8.0"新增cuda到環境變數:
1 |
|
在profile的最後新增:
1 2 3 |
|
使環境變數即時生效
1 |
|
驗證cuda是否安裝成功,輸入以下命令
1 |
|
顯示如下表示cuda安裝成功
1 2 3 4 |
|
嘗試編譯cuda提供的例子,下載cuda測試用例
1 |
|
1 |
|
看到類似如下圖片中的顯示,則代表CUDA安裝且配置成功
三、安裝cudnn6.0
到官網下載:cudnn-7.0-linux-x64-v3.0.8-prod.tgz,網址為: https://developer.nvidia.com/cudnn
1 2 3 4 |
|
四、安裝tensorflow-gpu
1 |
|
五、安裝pycharm用於python開發
1、安裝pycharm
1 2 3 4 5 6 7 |
|
2、安裝pycharm的jdk環境
1 2 3 4 5 |
|