由于已經(jīng)安裝了cpu版本了，如果再在該環(huán)境下安裝gpu版本會(huì)造成環(huán)境污染.
因此，再安裝gpu版本時(shí)，需要再新建一個(gè)虛擬環(huán)境才能安裝成功。
然后去官網(wǎng)下載所適配的版本。 安裝完cuda和cudnn后，開(kāi)始安裝pytorch的gpu版本。

1.安裝cude

首先查看windows電腦之前是否成功安裝了CUDA

第一步：同時(shí)按鍵盤(pán)上的“windows鍵+R”，輸入“cmd”并回車(chē)，進(jìn)入windows的命令行界面。

第二步：命令行里輸入“nvcc -V”并回車(chē)

第三步：如果已經(jīng)成功安裝CUDA的話，會(huì)顯示CUDA的版本號(hào)的。

已成功安裝cuda跳過(guò)cuda安裝部分

在計(jì)算機(jī)-管理-設(shè)備管理器-顯示適配器中，查看是否有獨(dú)立顯卡。

獨(dú)立顯卡NVIDIA GetForce GTX 1060

測(cè)試本機(jī)獨(dú)立顯卡是否支持CUDA的安裝（點(diǎn)擊此處）

本機(jī)的獨(dú)立顯卡是支持CUDA安裝

接下來(lái)就是選擇CUDA進(jìn)行安裝即可（點(diǎn)擊此處）

這里我選了CUDA Toolkit10.5的版本，至于選擇哪個(gè)版本，個(gè)人認(rèn)為應(yīng)該沒(méi)多大差別，一般就是看這個(gè)版本是否要求GPU的計(jì)算能力是多少以上。

根據(jù)電腦系統(tǒng)選擇相應(yīng)選項(xiàng)，exe選擇local，下載安裝

接著就是安裝過(guò)程，雙擊打開(kāi)顯示臨時(shí)解壓目錄，不需要改變，默認(rèn)即可。

接下來(lái)，進(jìn)入NVIDIA安裝過(guò)程，選項(xiàng)可參考其他博客。至于CUDA的安裝目錄，大家默認(rèn)安裝在C盤(pán)即可。（記住CUDA的安裝目錄）

安裝完成之后，便是配置環(huán)境變量。環(huán)境變量配置如下圖所示。（電腦->屬性->高級(jí)系統(tǒng)設(shè)置->環(huán)境變量->系統(tǒng)變量找到Path）

四個(gè)路徑分別對(duì)應(yīng)自己電腦安裝的相應(yīng)文件

測(cè)試CUDA是否正常安裝。可以看到測(cè)試成功

第一步：同時(shí)按鍵盤(pán)上的“windows鍵+R”，輸入“cmd”并回車(chē)，進(jìn)入windows的命令行界面。

第二步：命令行里輸入“nvcc -V”并回車(chē)

第三步：如果已經(jīng)成功安裝CUDA的話，會(huì)顯示CUDA的版本號(hào)的。

2.安裝cudnn
點(diǎn)擊此處，選擇下載download cudnn，但這里需要你注冊(cè)一個(gè)賬號(hào)，然后進(jìn)行問(wèn)卷之后才可以進(jìn)行下載頁(yè)面，反正一步步操作即可。

然后因?yàn)槲疑弦徊紺UDA的版本是11.5,而CUDNN的版本要跟CUDA版本一致

下載之后，解壓縮，將CUDNN壓縮包里面的bin、clude、lib文件直接復(fù)制到CUDA的安裝目錄下，直接覆蓋安裝即可。

3.安裝pytorch-gpu

①Anaconda安裝（之前安裝過(guò)的跳過(guò)）

Anaconda是一個(gè)用于科學(xué)計(jì)算的Python發(fā)行版，支持Linux、Mac和Window系統(tǒng)，提供了包管理與環(huán)境管理的功能，可以很方便地解決Python并存、切換，以及各種第三方包安裝的問(wèn)題。可以直接從 Anaconda官網(wǎng)（點(diǎn)擊此處）下載，但因?yàn)锳naconda的服務(wù)器在國(guó)外，所以下載速度會(huì)很慢，這里推薦使用清華的鏡像（點(diǎn)擊此處）來(lái)下載。選擇合適你的版本下載，這里選擇Anaconda2020.02Windowsx86_64.exe版本

下載之后，點(diǎn)擊安裝，按照提示依次點(diǎn)擊下一步，知道安裝結(jié)束。安裝完成后，在開(kāi)始菜單會(huì)出現(xiàn)按照后的軟件。打開(kāi)程序Anaconda Navigator，啟動(dòng)后可以看到Anaconda的環(huán)境界面

安裝完成后，進(jìn)行Anaconda的環(huán)境變量配置，電腦->屬性->高級(jí)系統(tǒng)設(shè)置->環(huán)境變量->系統(tǒng)變量找到Path，點(diǎn)擊編輯，加入三個(gè)文件夾的存儲(chǔ)路徑，然后選擇確定。點(diǎn)擊 Win+R 打開(kāi)，輸入cmd，打開(kāi)命令行，輸入conda info，顯示如下圖，說(shuō)明環(huán)境變量配置成功。

②創(chuàng)建虛擬環(huán)境（第一次安轉(zhuǎn)也可以選擇創(chuàng)建虛擬環(huán)境）

打開(kāi)anaconda prompt，首先創(chuàng)建虛擬環(huán)境：

#創(chuàng)建新環(huán)境
conda create --name pytorch-gpu python=3.6
#激活環(huán)境
conda activate pytorch-gpu

命令行輸入：conda create –name pytorch_gpu python=3.6
python_gpu為anaconda下虛擬環(huán)境名稱(chēng)，可自定義，python=3.6為選擇安裝的python版本。

proceed選擇y，回車(chē), 等待相關(guān)包下載，可以看到在安裝完成之后，信息提示。

#如果要啟用創(chuàng)建的環(huán)境，輸入指令

conda activate pytorch_gpu

#關(guān)閉該環(huán)境的話，輸入指令

conda deactivate

③安裝pytorch-gpu

激活后去pytorch官網(wǎng)（點(diǎn)擊此處）

選擇適合自己的版本，但是使用conda方式下載，直接用這個(gè)語(yǔ)句是不能下載成功的，因?yàn)樗姆?wù)器在國(guó)外，所以要選擇下國(guó)內(nèi)的鏡像源。（梯子在上面已經(jīng)給了鏈接）

如果沒(méi)有創(chuàng)建虛擬環(huán)境（第一次安裝），打開(kāi)Anaconda Prompt，前面是（base），可以直接在base里直接進(jìn)行安裝

如果創(chuàng)建了虛擬環(huán)境，需要激活虛擬環(huán)境，后再執(zhí)行condainstall

在已經(jīng)激活的環(huán)境中輸入上圖紅框中的命令：（可以直接選擇鏡像那個(gè)）

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

默認(rèn)pytorch官網(wǎng)為下載源，下載速度太慢，很容易報(bào)錯(cuò)

更改為清華大學(xué)鏡像，命令行輸入下面的命令：

# 添加Anaconda的清華鏡像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# 設(shè)置搜索時(shí)顯示通道地址
config --set show_channel_urls yes 

然后再輸入：

conda install pytorch torchvision torchaudio cudatoolkit=11.3

注意要把后面的-c pytorch去掉，不然還是使用的默認(rèn)源下載。

等待下載和安裝：

④測(cè)試

尋找到創(chuàng)建的python.exe（包含虛擬環(huán)境名的）進(jìn)行復(fù)制

D:\Users\Administrator\anaconda3\envs\pytorch-gpu\python.exe

運(yùn)行Pycharm，新建python文件輸入下面命令：

import torch
print(torch.__version__)  #注意是雙下劃線
print(torch.cuda.is_available())

選擇對(duì)應(yīng)的Python環(huán)境：

選擇剛剛創(chuàng)建的pytorch-gpu:（可直接測(cè)試一開(kāi)始復(fù)制的地址直接定位到）

最后運(yùn)行剛剛創(chuàng)建的Python文件，出現(xiàn)下面即是成功：

不報(bào)錯(cuò)，并返回True, 說(shuō)明安裝成功！！若返回False, 說(shuō)明系統(tǒng)的cuda版本號(hào)和 選擇的cuda版本不一致，需要將cuda版本進(jìn)行更新，具體更新操作如下：

右鍵-我的電腦-管理-設(shè)備管理器，找到顯示適配器，選擇NVIDIA，右鍵-更新驅(qū)動(dòng)程序，將更新到最新版本，重啟電腦即可。

注：如果要?jiǎng)h除自定義源更換回conda的默認(rèn)源，直接刪除channels即可，命令如下：

conda config --remove-key channels

總結(jié)

以上是生活随笔為你收集整理的安装Pytorch-gpu版本（第一次安装 或 已经安装Pytorch-cpu版本后）(win10系统安装教程)的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。