今天看了看pytorch官网竟然支持windows了,赶紧搞一个。
下载cuda 9.0 https://developer.nvidia.com/cuda-downloads
下载anaconda3 https://www.anaconda.com/download/
(记得安装在默认位置,不然的话windows下会发生很多错误)
pytorch官网查conda安装pytorch的命令
conda install pytorch -c pytorch
pip3 install torchvision (或者pip)
在conda中敲入python
>>import torch
>>torch.cuda.is_available
True
表示安装成功了
感觉比ubuntu简单啊,我的虚拟机可以删了.....