首頁>Club>
3
回覆列表
  • 1 # 易學智慧

    這裡以win10系統安裝gpu版tensorflow(cuda8.0+cudnn6.0)為例,為大家介紹AI開發環境的搭建。

    圖1-1

    2.然後選擇元件,可以看到NVCUDA.DLL後面的版本號即為需要下載的cuda版本,從下圖可以看到爪機支援的是CUDA8.0。

    圖1-2

    二.nvidia官網下載cuda8.0

    在網站https://developer.nvidia.com/cuda-toolkit-archive下載CUDA Toolkit 8.0 GA2

    圖2-1

    三. cuDNN6.0下載

    下載地址為https://developer.nvidia.com/cudnn,下載前要先註冊賬號(免費滴)。

    官網的推薦中,CUDA8.0對應cuDNNv7.2.1,但是tensorflow 1.4實測報錯要求cudnn6.0,所以還是下載6.0吧(如果確實要用cuDNNv7,可考慮編譯tensorflow原始碼)。

    四.安裝cuda 環境

    1.建議用管理員許可權執行下載的cuda.exe的檔案,開啟後會首先進行系統檢查

    下面繼續,同意後記得選擇自定義安裝:

    圖4-1

    2.這裡我選擇的是安裝前兩個,第三個驅動不需要,當然你們也可以嘗試裝一下

    圖4-2

    3.安裝位置建議選擇預設位置

    圖4-3

    圖4-4

    5.安裝完成成後需要去配置環境變數:

    對我的電腦右鍵依次選擇屬性,高階系統設定,環境變數,開啟如下介面。

    圖4-5

    6.然後重複新建以下變數:

    CUDA_PATH = C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0CUDA_PATH_V7_5 = C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0CUDA_LIB_PATH = %CUDA_PATH%\lib\x64CUDA_BIN_PATH = %CUDA_PATH%\binCUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64

    圖4-6

    7.依次填入後,找到path變數如圖,然後雙擊開啟:

    圖4-7

    圖4-8

    9.配置好之後可以去驗證一下是否已經裝好了cuda,在cmd視窗輸入nvcc -V(這裡我輸成了小寫v,所以報了個錯,注意是大寫V)

    小寫v報錯畫面:

    圖4-9

    大寫V正確畫面(這裡寫的是version,其實可以直接寫大寫的V就行了):

    圖4-10

    顯示版本,表明cuda8.0安裝成功了。

    五.安裝cuDNN

    將下載的cudnn解壓,可以看到裡面有三個資料夾:

    圖5-1

    將這三個資料夾解壓到:

    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

    (注意按照你自己實際的安裝位置來決定解壓位置)

    圖5-2

    六.開始安裝tensorflow-gpu 1.4.0

    建議大家先安裝一下Ananconda,他可以搭建虛擬環境,很好用,Anconda的安裝不再詳細去寫,網上很多,也很簡單,自己嘗試一下。

    安裝完成後開啟cmd視窗:

    1.建立conda環境,輸入:conda create -n gpu python=3.6(我的環境命名是gpu)

    圖6-1

    2.啟用進入你的conda環境:activate tensorflow-gpu

    3.出現的一些報錯(你們不一定有):

    no modle named "six" 解決方法:輸入 pip install six

    no modle named "pakging " 解決方法:輸入 pip install pakging

    no modle named setuptools.wheel 解決方法:pip install --upgrade setuptools

    或者 pip install setuptools

    4. 所有報錯解決後就可以pip install tensorflow-gpu==1.4.0

    圖6-2

    安裝完成了~

    七.測試

    最後可以開始測試了,各位只需要選擇你們建立的conda環境下的python.exe直譯器來執行即可,根據自己的編輯器去進行操作。

    當然也可以直接在這個環境下輸入cmd命令:python

    然後來進行個小測試吧:

    依次輸入下面內容

    import tensorflow as tf

    hello = tf.constant("Hello, TensorFlow!")

    sess = tf.Session()

    print(sess.run(hello))

    然後run起來

    圖7-1

    紅色部分為tensorflow輸出的log,主要是提醒(warning)。

    八.安裝cuda時的常見錯誤

    可能出現的情況一:

    圖8-1

    上圖表明情況正常,說明安裝的驅動比要求的新一點,直接點繼續

    圖8-2

    這種情況可能就是你的cuda版本跟系統不相容了,考慮換個cuda版本,根據我在上面給出的檢視顯示卡支援的cuda型號的方法去選擇。

    九.附錄:

    [1] https://blog.csdn.net/Candy_GL/article/details/79435151

  • 中秋節和大豐收的關聯?
  • 人到中年,失業瞭如何調整心態再就業?