通过Anaconda 安装
pytorch 是根据不同的cuda版本安装的
具体如下
cuda9.0
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
CUDA 8.x
conda install pytorch torchvision cudatoolkit=8.0 -c pytorch
CUDA 10.0
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
通过pip安装
python3
# Python 3.5
pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl
pip3 install torchvision