Pytorch安装

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.])    # 如正常则静默