前置軟件：Anaconda（其實(shí)也可以不用，如果你想裸奔的話）

如果有Anaconda了就不要單獨(dú)下在python編譯器了。

在cmd里面輸入一下代碼創(chuàng)建虛擬環(huán)境

解釋 ：創(chuàng)建環(huán)境名為 myenv 的虛擬環(huán)境，改虛擬環(huán)境的python版本為3.8.10

選擇y，確認(rèn)安裝。

之后我們輸入以下代碼

前面有（虛擬環(huán)境名）說明激活成功

下面我們查看電腦支持的CUDA版本，以我的電腦為例。

在桌面處右鍵打開NVIDIA 控制面板，點(diǎn)擊左下角的系統(tǒng)信息，然后點(diǎn)擊組件，得到以下界面

在此界面的NVCUDA.DLL可以看到此電腦支持的最高版本CUDA為10.1.

之后我們打開Pytorch的官網(wǎng)

因?yàn)樵诖私缑嫖覀冋也坏街С值?0.1的CUDA的安裝方法，所以我們點(diǎn)擊

Previous versions of PyTorch?

進(jìn)入界面選擇其他版本的CUDA安裝命令

我建議選擇pip開頭的命令，從圖片中我們可以找到命令（注意支持Windows還是Linux）。因?yàn)橹苯訌墓倬W(wǎng)下載的話，速度很慢，所以我們使用鏡像下載。

在這條命令的后面加上

也可以使用其他的鏡像

所以安裝的命令如下圖。記住要在激活虛擬環(huán)境的cmd中運(yùn)行。

驗(yàn)證

輸入python進(jìn)入編譯器，然后輸入以下代碼

import torch # 如果pytorch安裝成功即可導(dǎo)入

print(torch.cuda.is_available()) # 查看CUDA是否可用

print(torch.cuda.device_count()) # 查看可用的CUDA數(shù)量

print(torch.version.cuda) # 查看CUDA的版本號

以上我們完成了Torch的安裝