Win10下安装Pytorch深度学习环境

1 安装CUDA

• 去官网查询（适合的cuda版本）满足下图要求就行

适配显卡驱动版本查询地址


所以，cuda版本只和所装显卡驱动版本有关。以cuda8.0为例。

• 下载cuda8 地址
  
  双击安装即可，一直默认安装就行

打开终端输入

nvcc -V

看是否输出相应版本信息

• cudnn 官网下载地址
  下载cuda10.0支持的cudnn版本，解压后放入cuda安装根目录
  

2 Pytorch 安装

简单一句话用清华镜像原：真正的安装就两行命令，切换清华镜像原，安装
打开cmd
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch1.0.0 torchvision0.2.1 cuda80 (注意，上面找到的安装命令要去掉 -c pytorch)
完。