1 import torch 2 import torchvision 3 print(torch.cuda.is_available())
上面的命令只是检测CUDA是否安装正确并能被Pytorch检测到，并无说明是否能正常使用，要想看Pytorch能不能调用cuda加速，还须要简单的测试一下：测试
a = torch.Tensor(5,3) a=a.cuda()
print（a）
通常来说，输出主要是报48号错误，也就是CUDA的问题，出现这个问题在于硬件的支持状况，对于算力3.0的显卡来讲，若是安装了9.0的CUDA就会出现这个问题，解决的办法是退回CUDA8.0，或者更换更加高端的显卡，或者直接从源码编译，并在源码中作相应设置（修改setup.py文件里的TORCH_CUDA_ARCH_LIST，将这个值改为你当前使用的GPU对应算力！），对于最后一种方法，目前尚未尝试，不过，Pytorch确实作得不错，考虑的很周到~spa