1.更换下载镜像源为阿里云
pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/
2.到pytorch官网选择适合自己的版本
https://pytorch.org/
根据自己的配置去选择
第一项:pytorch的版本,stable为稳定版本
第二项:自己的系统
第三项:安装方式
第四项:语言
第五项:显卡的配置,这一项可参考
pytorch官网.png
3.测试是否安装成功?
1)如果安装的是GPU版本下面这几行测试是否可用CUDA(如果不能import torch ,那么一点是没装好)
import torch # 如正常则静默
import torchvision # 如正常则静默
print(torch.cuda.is_available())
2)如果安装的CPU版本用
import torch # 如正常则静默
a = torch.Tensor([1.]) # 如正常则静默