程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
您现在的位置: 程式師世界 >> 編程語言 >  >> 更多編程語言 >> Python

PyTorch0.4.1 GPU版本安裝完整步驟(Anaconda3 + Python3.6 + gpu版本,含CUDA9.2安裝、環境配置)

編輯:Python

目錄

一、下載並安裝cuda9.2

1. 首先查看電腦顯卡cuda版本,看最高支持的cuda為多少

2. 下載cuda9.2安裝包

3. 安裝cuda9.2

二、下載並安裝cudnn (for cuda9.2)

1. 下載cudnn

2. 安裝cudnn

三、配置環境變量

四、安裝GPU版本的PyTorch0.4.1

1. 准備:

2. 下載gpu版本的torch、torchvision的whl包(cu92):

3. 離線安裝gpu版本的torch、torchvision(cu92):


親測有用!!!

本人電腦:win10 + 有英偉達顯卡(GeForce GTX 1650) + Anaconda3 5.2.0 (其中Anaconda裡的Python版本是3.6),要安裝CUDA、PyTorch 0.4.1 版本(gpu版本)

目標:安裝gpu版本的PyTorch0.4.1(cuda9.2+pytorch0.4.1)

寫在前面:由於最近有需要,要安裝pytorch比較老的版本(零點幾的版本),恰好本人電腦帶有顯卡,所以想要直接安裝個gpu版本的pytorch古早版本。pytorch版本老,那麼相應的那個pytorch版本支持的cuda版本也比較老,所以我接下來就打算:1)先裝個cuda9.2(大致有cuda8.0、cuda9.0、cuda9.1、cuda9.2、cuda10.0等等很多版本),並安裝好對應的cudnn,如果到時候cuda9.2太老的話,可以再換裝其他版本。。。。。2)安裝好cuda9.2和cudnn9.2之後,然後再安裝一個gpu版本(cu92,和前面的cuda9.2對應)的PyTorch0.4.1版本。(本來打算安裝PyTorch0.4.0之前的版本,但一是以前版本的安裝包還挺少的,二是以前版本對應的gpu版本下匹配的cuda版本也很少很老,選擇的余地比較少。所以就先打算安裝cu92/PyTorch0.4.1試試水,不行的話可以再換裝其他。)

一、下載並安裝cuda9.2

1. 首先查看電腦顯卡cuda版本,看最高支持的cuda為多少

在命令行通過命令nvidia-smi查看cuda版本,如下圖。可以看到我的顯卡最高支持cuda11.2,即可以裝cuda10.0、cuda11.0等,但不能裝cuda11.3及以上。(查看顯卡和cuda版本可以參考之前博客:win10查看Nvidia顯卡、cuda版本)

2. 下載cuda9.2安裝包

cuda官網下載地址:CUDA Toolkit Archive | NVIDIA Developer

點擊上面的官網下載鏈接,選擇cuda9.2,點進去。(此處選擇下載cuda9.2,是一個大約1.5G的exe文件。)

  • 也可以使用百度雲下載cuda9.2和對應的cudnn9.2,百度雲下載鏈接: CUDA9.2和配套的cuDNN百度雲直達下載,提取碼:81v4。來自博客(pytorch gpu版本的最全安裝教程,含環境配置、CUDA(百度雲下載)安裝程序失敗解決方案)

然後選擇相應的選項(windows、x86_64、win10、exe[local]),點擊download即可。網上很多博客強調此處一定要選擇exe[local]離線安裝,不然在線安裝包很坑,下載巨慢。。。

cuda9.2下載下來後是一個exe安裝包。

3. 安裝cuda9.2

雙擊運行上面下載好的cuda9.2的exe文件,直接選擇默認的C盤安裝路徑(盡量不要自定義路徑,避免後面配置環境變量出現不必要的麻煩)。

 選擇自定義安裝:

只選擇第一個選項CUDA即可,其他的不需要勾選。(參考了網上其他博客,很多建議第一個CUDA裡的"Visual Studo Integration"最好也不要勾選,CUDA安裝失敗一般都是因為它導致的。但我電腦裡之前安裝過VS2015,所以此處勾選了"Visual Studo Integration",應該問題不大,後面我也確實安裝成功了,沒遇到什麼問題。所以,如果勾選這個選項的話,最好要先安裝一下VS,然後再來安裝cuda。)

後面的過程都是直接點擊"下一步“,即可。

最後cuda9.2安裝成功!!!沒遇到什麼問題。 

二、下載並安裝cudnn (for cuda9.2)

1. 下載cudnn

cudnn官網下載地址:cuDNN Archive | NVIDIA Developer

下載完CUDA9.2之後,在cudnn官網下載鏈接找與CUDA9.2版本匹配的cuDNN,選最新的安裝即可。(官網下載好像需要注冊登錄。 可以直接使用上面cuda9.2下載時提供的百度雲鏈接,裡面有cuda9.2對應的cudnn9.2,這樣就不需要自己在官網下載了。)

 下載好的cudnn壓縮包:

2. 安裝cudnn

將上面下載的cudnn壓縮包解壓縮,然後分別將解壓後的bin、include、lib\x64三個文件夾中的.dll、.h和.lib文件復制到CUDA安裝目錄下的對應文件夾裡,即可。(注意是復制過去,而不是直接覆蓋!!!我試了一下,直接把三個文件夾拉到左邊路徑下,會直接把對應文件復制到對應文件夾下。)

三、配置環境變量

1)計算機上點右鍵,打開屬性->高級系統設置->環境變量,可以看到系統中已經自動多了下面兩個環境變量。

2)接下來,還要繼續在系統裡添加幾個環境變量:

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

3)添加完環境變量之後,在cmd命令行裡面使用命令nvcc -V進行測試,測試失敗。。。。顯示nvcc不是內部或外部命令。。。無法識別命令。。。(使用下面參考博客裡的方法,最終ncvv -V測試成功!!!)

解決方法:

解決nvcc顯示不是內部或外部命令的問題_Lonely Planet.的博客-程序員秘密_nvcc' 不是內部或外部命令 - 程序員秘密

打開“cmd”,輸入set PATH=c:  ,然後輸入echo %PATH%,關閉“cmd”,之後再次開啟命令提示符,輸入echo %PATH%,就會發現新添加的環境變量已經生效了,再次輸入ncvv -V,成功。 

 

四、安裝GPU版本的PyTorch0.4.1

1. 准備:

1)首先安裝Anaconda。(很簡單,自行搜索)

2)其次在用Anaconda建一個虛擬環境。

打開Anaconda Prompt輸入:conda create -n pytorch0.4.1 python=3.6(n表示參數name,pytorch0.4.1表示自己起的環境的名字,python=3.6表示需要使用的一些包。)

conda create -n pytorch0.4.1 python=3.6

3)激活虛擬環境:

conda activate pytorch0.4.1 

2. 下載gpu版本的torch、torchvision的whl包(cu92):

torch和torchvision的whl包下載鏈接:http://download.pytorch.org/whl/torch_stable.html(根據cuda版本和python版本下載相對應的pytorch版本。)

下載2個whl包:

  • cu92/torch-0.4.1-cp36-cp36m-win_amd64.whl(cuda9.2,torch0.4.1,cp36代表python版本是3.6,根據自己的需要下載。)
  • cu92/torchvision-0.4.1+cu92-cp36-cp36m-win_amd64.whl(cuda9.2,torchvision-0.4.1)

此處,下載了2個whl包:cuda9.2相應的torch0.4.1、cuda9.2的torchvision0.4.1。(根據torch和torchvision版本兼容匹配表https://github.com/pytorch/vision#installation ,torch0.4.1應該匹配torchivision0.2.2版本,但是cuda9.2相應的torchvision包沒有0.2.2版本的,於是我下載了cuda9.2/torchivision0.4.1版本,感覺應該問題不大。。。不過最後測試的時候還是改回了torchvision0.2.2版本。。。)

 

將whl包下載到D盤下,如下圖所示:

3. 離線安裝gpu版本的torch、torchvision(cu92):

在cmd命令行轉到torch和torchvision的whl包保存目錄,pip install whl包進行離線安裝,即可。

1)安裝gpu版本(cu92)的pytorch0.4.1:

pip install torch-0.4.1-cp36-cp36m-win_amd64.whl

 

2)安裝gpu版本(cu92)的torchvision0.4.1: 

pip install "torchvision-0.4.1+cu92-cp36-cp36m-win_amd64.whl"

 3)測試:

import torch

torch.__version__

torch.cuda.is_available()

但是測試torchvision時出現錯誤。。。 錯誤如下圖所示。。。可能主要是因為torch0.4.1和torchvision0.4.1的版本不匹配,根據版本兼容表,torch0.4.1版本應該匹配torchvision0.2.2版本。。。。於是,打算重新安裝torchvision0.2.2。

如下圖,重新安裝torchvision0.2.2。這次,測試成功了。(不過不知道以後會不會出現什麼問題,如果有問題,那就根據自己的情況,再重新安裝需要的版本吧。)

至此,gpu版本的PyTorch0.4.1安裝完成!!!(cuda9.2+pytorch0.4.1)

參考:

  • (帶你避坑)win10安裝帶CUDA的Pytorch看這篇就夠了
  • pytorch gpu版本的最全安裝教程,含環境配置、CUDA(百度雲下載)安裝程序失敗解決方案
  • GPU版本pytorch安裝教程
  • win10下pytorch-gpu安裝以及CUDA詳細安裝過程
  • windows10安裝GPU版pytorch詳細教程(以cuda10.0+pytorch1.1為例,手把手教學)
  • windows10安裝GPU版pytorch步驟(全程離線)

  1. 上一篇文章:
  2. 下一篇文章:
Copyright © 程式師世界 All Rights Reserved