1. 程式人生 > >pytorch指定版本安裝及檢視版本

pytorch指定版本安裝及檢視版本

  • pytorch官網
https://pytorch.org/
  • pytorch舊版本網址
https://pytorch.org/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__)