pip3安裝pytorch
1、進入官網下載相應的pytorch
2、離線下載輪子
可以直接復制Run this Command中的命令進行安裝,但由於安裝速度實在太慢,更換了源亦是如此。所以,為了加快安裝速度,事先下載wheel文件,進行離線安裝。
瀏覽器打開上圖紅框網站。
使用迅雷下載,對應的版本。
3、使用pip3安裝whl文件
4、安裝pytorch
由於已經安裝好whl文件,所以再次使用pip3命令安裝將非常快。
pip3 install torch===1.7.1 torchvision===0.8.2 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
注意,在此步驟中會安裝一些numpy、feature等packet,會影響一些安裝速度,甚至出現一些錯誤。所以,可以選擇提前安裝這些packet。(提示缺什么安裝什么)
5、測試
import torch if __name__ == '__main__': print(torch.__version__) print(torch.cuda.is_available())
1.7.1
True
Process finished with exit code 0