https://blog.csdn.net/qq_38424674/article/details/109499015

还需要安装torchvision-0.4.0-cp36-cp36m-win_amd64.whl以配合torch1.2.0  https://download.pytorch.org/whl/cu100/torch_stable.html

然后 pip install xxx.whl .....就可以啦

 

 

检查是否可用

import torch
#返回当前设备索引

print(torch.cuda.current_device())

#返回GPU的数量
print(torch.cuda.device_count())

#返回gpu名字，设备索引默认从0开始
print(torch.cuda.get_device_name(0))

#cuda是否可用
print(torch.cuda.is_available())