pytorch官网

https://pytorch.org/

pytorch旧版本网址

https://pytorch.org/get-started/previous-versions/
pytorch从链接安装指定版本

pip install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl

注:上述版本为cuda 8.0,python 2.7,torch 0.2.0版本

torchvision安装

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torchvision
# 验证
import torch
import torchvision
print(torch.cuda.is_available())
# pytorch版本查看
import torch
print(torch.__version__)

更多推荐