python 3.8.6 + pytorch 1.7.0 cuda10.2 pip安裝踩坑


如果是直接按照官網給出的命令安裝,有可能會提示not found,原因未知。
這里給出的解決方法是直接訪問末尾給出的鏈接https://download.pytorch.org/whl/torch_stable.html
手動下載cu102/torch-1.7.0-cp38-cp38-win_amd64.whl,和cu102/torchvision-0.8.0-cp38-cp38-win_amd64.whl
如果電腦下載速度慢,可嘗試眾所周知的方法,或者用手機連wifi下載(這個是玄學了)
然后通過命令行安裝,比如

pip install C:\torchvision-0.8.0-cp38-cp38-win_amd64.whl

cuda 10.2需要按照其他教程配置完畢。
以下是pytorch gpu測試代碼

import torch
print(torch.__version__)
torch.cuda.is_available()


免責聲明!

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



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