當前位置:
首頁 > 最新 > 在Window10下安裝Anaconda3和tensorflow-gpu

在Window10下安裝Anaconda3和tensorflow-gpu

【簡述】

關於Window安裝TensorFlow- GPU環境的文章網上已經有很多了,但是為什麼還要寫這篇文章呢,就是被網上的文章給坑了。由於tensorflow-gpu的版本和CDUDA版本,甚至CUDNN版本都有著對應關係,導致在安裝TensorFlow- GPU入了很多坑,最後還是通過看錯誤信息,才順利安裝上的,因此分享這篇文章,避免小夥伴們入坑(大家可以對照我的模式來安裝)。

借鑒了兩位大神的安裝教程之後,綜合一下終於安裝成功了。

https://www.cnblogs.com/raorao1994/p/8857229.html

https://blog.csdn.net/qilixuening/article/details/77503631

先上環境:

【詳細過程】

一、前提準備條件(所有內容全為64位,不要問為什麼)

1、Python-3.5

2、Anaconda3-5.1.0 (python 3.6 version,64位)

3、顯卡驅動:建議去英偉達官方下載,驅動不要最新的,時間最好跟cuda和cudnn發布時間相差無幾。

4、CUDA-9.0.176_win10 (是V9.0版本)

此外包括了cuda9.0.176的兩個補丁CUDA-9.0.176.1_windows 和 CUDA-9.0.176.2_windows

5、CUDNN-9.0-windows10-x64-v7 (其實是V7.0,官網下載下來是這個,需要註冊帳號)

二、安裝過程

1、安裝python-3.5,默認安裝即可,記得勾選環境變數設置。

2、Anaconda3的安裝,也默認安裝,記得勾選環境變數設置。

3、安裝英偉達顯卡驅動,默認精簡安裝即可。

4、CUDA的安裝(核心來了)

① 右鍵以管理員身份運行cuda_9.0.176_win10.exe,下面按照圖片操作

以上便是安裝完畢cuda的主程序。

② 接下來安裝兩個補丁,也分別以管理員身份運行,先安裝補丁1:CUDA-9.0.176.1_windows,詳細如下圖所示:

③再安裝補丁2:CUDA-9.0.176.2_windows,詳細如下圖所示:

以上便是cuda主程序和兩個補丁的完整安裝教程。

5、CUDNN的安裝

解壓CUDNN-9.0-windows10-x64-v7.zip,壓縮包裡面cuda文件下有三個文件夾(bin,include,lib)和一個txt文件,把這三個文件夾複製到你cuda的安裝目錄下,如果按照上面默認安裝的話,路徑是C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0,將文件複製過來即可。

6、環境變數的設置 ★★★★★★

接下來便是核心的核心了,網上教程大部分都是沒有這部分,從而導致無法調用GPU。Window10系統直接小娜搜索「環境變數」,運行「編輯系統環境變數」,然後選擇「高級」→「環境變數」,我們看到「系統變數」已經有 CUDA_PATH 和 CUDA_PATH_V9_0 這兩個環境變數,還需要添加以下幾個環境變數。

① 在「系統變數」下面點擊「新建」,在彈出的窗口「編輯系統變數」中輸入變數名和變數值,然後確定。

「變數名」:CUDA_SDK_PATH

「變數值」:C:ProgramDataNVIDIA CorporationCUDA Samplesv9.0

② 再依次新建以下四個變數名和變數值,分別確定保存。

CUDA_LIB_PATH = %CUDA_PATH%libx64

CUDA_BIN_PATH = %CUDA_PATH%in

CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%inwin64

CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%commonlibx64

③ 在系統變數「Path」增加以下幾條路徑,點擊「Path」,選擇「編輯」。

在「編輯環境變數」窗口點擊「新建」,依次新建以下四個變數

%CUDA_LIB_PATH%

%CUDA_BIN_PATH%

%CUDA_SDK_LIB_PATH%

%CUDA_SDK_BIN_PATH%

再依次通過「新建」添加如下4條(默認安裝路徑):

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0libx64;

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0in;

C:ProgramDataNVIDIA CorporationCUDA Samplesv9.0commonlibx64;

C:ProgramDataNVIDIA CorporationCUDA Samplesv9.0inwin64;

如果你選用了自定義路徑,上述這些默認路徑都應該相應替換為你的自定義路徑。安裝完畢後如下圖所示:

到此,環境變數配置完畢,我們可以驗證是否配置成功,主要使用CUDA內置的deviceQuery.exe 和 bandwithTest.exe:首先win+R啟動cmd,cd到安裝目錄:C:Program FilesNVIDIA GPU Computing TookitCUDAv9.0extrasdemo_suite,然後分別執行bandwidthTest.exe和deviceQuery.exe,執行完畢,兩個都返回「Result=PASS」,那麼就證明環境變數配置成功。

三、安裝tensorflow-gpu

Win+R啟動cmd,在命令提示符內輸入以下

C:User***>conda creat –n tensorflow-gpu python=3.6

(這樣在anaconda建議一個名字叫tensorflow-gpu的環境)

C:User***>activate tensorflow-gpu

(激活tensorflow-gpu環境)

C:User***>pip install tensorflow-gpu==1.7.0

(安裝tensorflow-gpu 1.7.0版)

待tensorflow-gpu安裝完畢後,基於window10+anaconda3+tensorflow-gpu版就全部安裝完畢了,可以採用網上各種方法測試顯卡的威力。

【總結】

經過兩天的不斷嘗試,最後總算安裝完畢並成功在window10下用GPU運行程序,下圖是測試效果,總的來說,環境變數一定要嚴格設置才能避免各種錯誤。

測試前:

測試中:

註:數據部分使用馬賽克加以遮擋。

至此整套在Window10下安裝Anaconda3和tensorflow-gpu流程結束。


喜歡這篇文章嗎?立刻分享出去讓更多人知道吧!

本站內容充實豐富,博大精深,小編精選每日熱門資訊,隨時更新,點擊「搶先收到最新資訊」瀏覽吧!


請您繼續閱讀更多來自 全球大搜羅 的精彩文章:

有了這份劇單,我只想躺上三天三夜
我是學渣可我不是垃圾

TAG:全球大搜羅 |