官网-- 安装torch1.1.0的地址
https://pytorch.org/get-started/previous-versions/
找到安装CUDA9.0版本的代码
# CUDA 9.0
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
直接用上面的代码会发现这样是安装不了的,这是个巨坑,估计是国外镜像的原因
换成pip安装,首先要修改pip镜像:清华镜像配置
# 更新pip
python -m pip install --upgrade pip -i
# 安装torch1.1.0
pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl
# 安装 torchvision0.3.0
pip install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp36-cp36m-win_amd64.whl
安装成功。
测试torch是否支持gpu
import torch
# 查看torch的版本号
print(torch.__version__)
# 是否支持gpu
print(torch.cuda.is_available())
如果显示True则支持gpu,如果显示Flase 不支持GPU