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確實做得不錯，考慮的很周到~