如果是直接按照官網給出的命令安裝,有可能會提示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()
