PyTorch就是一個計算工具,你能夠借助它,使計算機完成復(fù)雜的計算。

Py就是Python,Torch字面意思是火炬,我們可以把它理解成能在GPU中計算的矩陣,而且運算速度特別快。

為了能在后續(xù)的課程中,更好的學(xué)習(xí)Pytorch,本小節(jié)我們來學(xué)習(xí)如何安裝它。

3.4.1?下載與安裝

首先訪問pytorch官網(wǎng)（pytorch.org）,點擊install

?

先選擇pytorch的版本,這里我們選擇最新的穩(wěn)定版1.11; 接著是操作系統(tǒng),我這里使用的是windows;安裝方式 我使用conda進行安裝; 編程語言選擇python;

計算平臺?我選擇新版的Cuda11.3

這里如果你對conda 和 cuda還不了解,建議你先看一下這一章的前兩節(jié)教程。

我來捋一下?pytorch cuda 與 Nvidia之間的關(guān)系：我們想安裝pytorch ，使用GPU進行快速運算，那么就需要先安裝CUDA，因為它是用來管理GPU實現(xiàn)并行運算的。而CUDA toolkit是Nvadia公司的推出的，所以安裝cuda之前，需要確保你的電腦有英偉達獨立顯卡。

同時要注意下載的版本問題，要根據(jù)顯卡驅(qū)動的版本，下載支持的Cuda 版本。如果有必要，可以到這個地址（https://www.nvidia.com/Download/index.aspx?lang=cn）?先升級你的顯卡驅(qū)動，再根據(jù)Cuda 版本安裝支持的Pytorch版本。

如果你的電腦沒有英偉達獨立顯卡，那么安裝CPU版本也是可以的。在學(xué)習(xí)階段，使用CPU也能支持跑個小實驗。但是你要知道實際項目中還是使用GPU來訓(xùn)練模型。

然后復(fù)制下面這條生成好的命令

打開anaconda prompt，使用conda創(chuàng)建一個新的虛擬環(huán)境，用來接下來pytorch的學(xué)習(xí)：

conda create -n pytorch python=3.8

-n 后面是虛擬環(huán)境的名字 ，我這里叫pytorch ，當(dāng)然你也可以叫別的，python的版本指定了3.8

環(huán)境創(chuàng)建好后，使用：conda activate pytorch 激活這個環(huán)境，可以看到此時括號里就是pytorch

?

接下來，把從網(wǎng)站上復(fù)制的安裝命令，粘貼在這，回車。剩下的就是等待整個安裝過程。它會安裝pytorch以及其關(guān)聯(lián)的包。

?

如果要安裝之前版本的Pytorch，請訪問INSTALLING PREVIOUS VERSIONS OF PYTORCH,找到要安裝的版本，進行安裝。

3.4.2?檢測是否安裝成功

PyTtorch安裝完成后，需要驗證是否安裝成功。

先在這個環(huán)境中啟動python：輸入python，回車

然后導(dǎo)入torch模塊：import torch 回車.?注意是torch而不是pytorch

如果像我這樣，界面沒有報任何的錯誤信息，那么意味著PyTorch已經(jīng)順利安裝了

?

我們來創(chuàng)建一個torch張量，輸入：x = torch.rand(5) ，打印x，也可以正常工作

現(xiàn)在我們來檢查CUDA，也就是GPU是否可用

輸入：torch.cuda.is_available() 回車

如果輸出的是True，意味著可以使用GPU，大功告成

?

如果你安裝的明明是?GPU 版本，但是返回的卻是“False”，GPU 不可用。那么你就要對照著上面講解pytroch cuda 與 nvidia關(guān)系的地方檢查，一定可以找到原因的。

?

同步更新：

Github: https://github.com/Gengzhige

擴展學(xué)習(xí)：

https://www.bilibili.com/cheese/play/ss1380?spm_id_from=333.999.0.0

?