Win10系统安装Torch的方法

我选择的是Win10+Cuda9.0+Python3.7的安装

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

如果想选择pip3下载的话

pip3 install https://download.pytorch.org/whl/cu90/torch-1.0.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision

import torch
from torch.backends import cudnn
a = torch.Tensor([1.0])
a = a.cuda()
print(a)

# CUDNN TEST
from torch.backends import cudnn
print(cudnn.is_acceptable(a))