13、pip3安裝pytorch


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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM