在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:全球大搜羅 |