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

2021-06-07 18:04

阅读:538

标签:tab   imp   原因   下载速度   通过命令   tor   踩坑   amd   配置   

如果是直接按照官网给出的命令安装,有可能会提示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()

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

标签:tab   imp   原因   下载速度   通过命令   tor   踩坑   amd   配置   

原文地址:https://www.cnblogs.com/weypro/p/14552225.html


评论


亲,登录后才可以留言!