Mac和window安装GPU版pytorch1.1.0

官网-- 安装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