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())