我选择的是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))