ubuntu16.04安装torch torchvision
1.安装torch0.4.1
Via pip
找到对应的网站下载whl文件
#CPU-only build
https://download.pytorch.org/whl/cpu/torch_stable.html
#CUDA 8.0 build
https://download.pytorch.org/whl/cu80/torch_stable.html
#CUDA 9.0 build
https://download.pytorch.org/whl/cu90/torch_stable.html
#CUDA 9.2 build
https://download.pytorch.org/whl/cu92/torch_stable.html
#CUDA 10.0 build
https://download.pytorch.org/whl/cu100/torch_stable.html
下载好后开始安装
pip3 install torch-0.4.1-cp36-cp36m-linux_x86_64.whl
2.安装torchvision0.2.1
pip3 install torchvision==0.2.1
3.测试版本
python3
import torch
import torchvision
torch.__version__
torchvision.__version__