pytorch查看CUDA支持狀況，只須要三行代碼，另附Cuda runtime error (48) : no kernel image is available for execution處理辦法

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