Pytorch安装 Win10 + Python 3.6 + Anaconda + CUDA 10.0

1. 确认CUDA-enabled GPU

Win + R: dxdiag 查看显卡类型
在NVIDIA官网查看是否支持CUDA

2. 安装NVIDIA Driver和CUDA Toolkit

NVIDIA Driver Downloads
https://www.nvidia.com/Download/index.aspx?lang=en-us

CUDA Toolkit 10.0 Archive
https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

官方安装教程（包括网络安装版本/network和本地安装版本/local)
https://docs.nvidia.com/cuda/cuda-quick-start-guide/index.html#windows

注意 安装CUDA前需要装好可用版本的Visual Studio，我本来装的是Visual Studio 2019，但是在安装过程中报错检测不到Visual Studio，在装完Visual Studio 2017以后问题解决
Visual Studio 2017 下载

3. 安装Pytorch

Pytorch官网 https://pytorch.org/get-started/locally/
（不知道为什么中英文官网好像有一些微妙的差别）

可以直接选择配置用conda安装

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

但是我用conda一直报错找不到channel
PackagesNotFoundError: The following packages are not available from current channels

可以尝试https://www.cnblogs.com/hellojiaojiao/p/10790273.html / https://blog.csdn.net/miao0967020148/article/details/85230430 提供的解决方法

但我改用了看起来更简单的pip install

CUDA 10.0

pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1-cp36-cp36m-win_amd64.whl 
pip3 install torchvision

但这样安装以后我import pytorch还是没有成功 发现是因为pip3会把包装到自带的3.x版本的默认环境 (参考https://blog.csdn.net/qq_22526061/article/details/81040778） 在我的环境（py=3.6）中 pip就是pip3 所以直接用pip就可以

pip install https://download.pytorch.org/whl/cu100/torch-1.0.1-cp36-cp36m-win_amd64.whl 
pip install torchvision 

4. 验证Pytorch

参考官网

from __future__ import print_function
import torch
x = torch.rand(5,3)
print(x)

结果会输出一个tensor

进一步验证CUDA

import torch
torch.cuda.is_available()

运行时报错
RuntimeError: module compiled against API version 0xb but this version of numpy is 0xa

原来是因为numpy版本问题（参考https://blog.csdn.net/chris_zhangrx/article/details/78407459)更新numpy

pip install --upgrade numpy

结束