参考:https://www.cnblogs.com/GrPhoenix/p/10324172.html
1.Anaconda已经安装完毕
2.创建python虚拟环境:conda create -n pytorch-gpu python=3.6
3.激活创建的虚拟环境:activate pytorch-gpu
4.安装pytorch: https://pytorch.org/get-started/previous-versions/ :conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
5.如果安装太慢,可以先下载到本地:torch-1.2.0-cp36-cp36m-win_amd64.whl
6.安装pip install torch-1.2.0-cp36-cp36m-win_amd64.whl
7.测试:
>>> import torch
>>> torch.cuda.is_available()
返回true,这GPU版本的可用。