Win10:在Anaconda中安装cuda+pytorch

1 创建虚拟环境

conda create -n pytorch_gpu python=3.6

备注：修改虚拟环境名称的方法

conda create --name python2（新名称） --clone python1（老名称）

conda remove --name python1（老名称） --all 

2 激活虚拟环境

activate pytorch_gpu

3 安装tensorflow-gpu的过程中会自动适配安装cuda以及cudnn

conda install tensorflow-gpu

4 查看CUDA版本

conda list

可从中查找到类似如下内容，CUDA版本为10.1：

5 安装对应CUDA版本的pytorch

参考网址：https://pytorch.org/

conda install pytorch torchvision cudatoolkit=10.1

6 检验

在该虚拟环境下依次输入如下指令：

python

import torch

print(torch.cuda.is_available())

若输出为 True 则表明安装成功。