ubuntu16.04安装torch torchvision

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__

你可能感兴趣的:(python)