python3.6環境配置流程(gpu版本)


  Tensorflow有兩個版本:GPU和CPU版本,CPU的很好安裝;GPU 版本需要 CUDA 和 cuDNN 的支持,如果獨顯+集顯,那么推薦你用GPU版本的,因為GPU對矩陣運算有很好的支持,會加速程序執行!並且CUDA是Nvidia下屬的程序,所以GPU最好是Nvidia的,AMD的顯卡沒有CUDA加速!滿足以上條件之后,你需要查看一下你的英偉達GPU是否支持CUDA,如下圖:(官網鏈接:CUDA GPUs | NVIDIA Developer

 

滿足條件后可以進行安裝。

第一步:安裝Anaconda(已安裝)

1.下載安裝軟件

2.配置環境變量

  計算機右鍵——屬性——高級系統設計——高級——環境變量(也可以右下角直接搜索)

點擊環境變量

選擇“Path”,點擊“編輯”

將以下三個路徑加入,注意這里要換成你自己的安裝路徑,我的在C盤。

  • C:\Anaconda3
  • C:\Anaconda3\Scripts
  • C:\Anaconda3\Library\bin

 

然后點擊“確定”保存,在左下角打開Anaconda—Anaconda Prompt,輸入“conda -V”,能正常顯示版本號,證明已經配置好了。

 

 

 

第二步:安裝CUDA Toolkit + cuDNN

1.查看需要安裝的CUDA+cuDNN版本

注意,tensorflow是在持續更新的,具體安裝的CUDA和cuDNN版本需要去官網查看,要與最新版本的tensorflow匹配。

點擊查看最新tensorflow支持的CUDA版本:https://www.tensorflow.org/install/install_windows#requirements_to_run_tensorflow_with_gpu_support

官網對於tensorflow和CUDA以及cuDNN都有更新,並不是最新的才可以用,關鍵是匹配好對應的版本

2.下載CUDA + cuDNN

在這個網址查找CUDA已發布版本:https://developer.nvidia.com/cuda-toolkit-archive

根據提示,我下載11.2版本

進入下載界面進行下載

 

 

 

 

 

 

下載好CUDA Toolkit 9.0 后,我們開始下載cuDnn 7.0,需要注意的是,下載cuDNN需要在nvidia上注冊賬號,使用郵箱注冊就可以,免費的。登陸賬號后才能下載。

(其實查詢網上資料不注冊也可以下載,直接右擊,需要下載的鏈接,在迅雷中就可以了,大家可以嘗試一下)

cuDNN歷史版本在該網址下載:https://developer.nvidia.com/rdp/cudnn-archive

注意版本號要與tensorflow官網要求完全相同,我這里也為11.2版本

下載win10版本

這樣,我們就下載好了 CUDA Toolkit 11.2.0 和 cuDnn 8.1.0.77,下面我們開始安裝。

3.安裝 CUDA Toolkit 11.2.0和 cuDnn 8.1.0.77

至關重要的一步:卸載顯卡驅動

由於CUDA Toolkit需要在指定版本顯卡驅動環境下才能正常使用的,所以如果我們已經安裝了nvidia顯卡驅動(很顯然,大部分人都安裝了),再安裝CUDA Toolkit時,會因二者版本不兼容而導致CUDA無法正常使用,這也就是很多人安裝失敗的原因。而CUDA Toolkit安裝包中自帶與之匹配的顯卡驅動,所以務必要刪除電腦先前的顯卡驅動。

再進行安裝

安裝

此處選擇“自定義(高級)”

勾選所有

一路通過即可。

接下來,解壓“cudnn-11.2-windows-x64-v8.1.0.77.zip”,將一下三個文件夾,拷貝到CUDA安裝的根目錄下。

這樣CUDA Toolkit 11.2.0 和 cuDnn 8.1.0就已經安裝了,下面要進行環境變量的配置。

配置環境變量

將下面四個路徑加入到環境變量中,注意要換成自己的安裝路徑。

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

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

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64

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

到此,全部的安裝步驟都已經完成,這回我們測試一下。

第二步:安裝TensorFlow-GPU

打開tensorflow官網:https://www.tensorflow.org/install/install_windows#installing_with_anaconda

 

跟着操作步驟走就可以了。

1.創建conda環境

通過調用下列命令,創建一個名為“tensorflow”的conda環境:

conda create -n tensorflow pip python=3.6

等待相應包的安裝,如果國內網絡太慢的話,可以為conda設置清華源,這樣速度能快一點,具體配置過程,網上查一下吧,此處不再講述。如果看到這樣的提示,就證明conda環境創建成功。

2.激活環境

通過以下命令激活conda環境:

activate tensorflow

這樣就進入了剛創建的“tensorflow”環境。

3.安裝tensorflow-gpu

安裝GPU版本的tensorflow需要輸入以下命令(這里我裝的2.5.0版本,keras-nightly也是2.5.0版本):

pip install --ignore-installed --upgrade tensorflow-gpu 

如果只需要安裝CPU版本的tensorflow則輸入以下命令:

pip install --ignore-installed --upgrade tensorflow

這樣就安裝成功了。

注意:務必注意一點,在安裝完tensroflow后,由於我們是新創建的conda環境,該環境中基本上是空的,有很多包和IDE並沒有安裝進來,例如“Ipython”,“spyder”此時如果我們在該環境下打開spyder/Ipyton/jupyter notebook等,會發現其實IDE使用的kernel並不是新建立的這個環境的kernel,而是“base”這個環境的,而“base”環境中我們並沒有安裝tensorflow,所以一定無法import。這也就是為什么有很多人在安裝好tensorflow后仍然在IDE里無法正常使用的原因了。

 

第四步:測試

 

測試一下是否能import tensorflow,但是一定要進入到我們的tensorflow環境中,不要在base環境下,激活tensorflow環境

 

運行出b'Hello tensorflow'證明我們的tensorflow已經安裝成功了。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM