PyTorch的安裝非常簡單,分為兩步:
(1)安裝Anaconda
這個網(wǎng)上教程很多谒主,這里不再詳述筐高。
(2)安裝PyTorch
打開PyTorch官網(wǎng),找到下圖所示位置泡嘴,選擇自己需要的環(huán)境:
image.png
會自動生成conda的安裝命令甫恩,如下所示:
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
把這個命令復(fù)制到cmd命令行執(zhí)行安裝即可。
驗證PyTorch是否安裝成功:
import torch
print(torch.__version__)
如果導(dǎo)入成功酌予,并且能輸出正確的版本信息磺箕,就說明pytorch安裝成功:
'1.1.0'
(3)使用pip方式安裝(備選)
使用conda安裝pytorch如果非常慢,可以使用pip方式安裝抛虫,如下圖所示:
image.png
pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp37-cp37m-win_amd64.whl
pip3 install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp37-cp37m-win_amd64.whl
根據(jù)命令中的地址松靡,手動下載這兩個安裝文件到本地,如E:\盤
E:\torch-1.1.0-cp37-cp37m-win_amd64.whl
E:\torchvision-0.3.0-cp37-cp37m-win_amd64.whl
使用pip3命令安裝即可:
pip3 install E:\torch-1.1.0-cp37-cp37m-win_amd64.whl
pip3 install E:\torchvision-0.3.0-cp37-cp37m-win_amd64.whl