PyTorch查看CUDA支持情况

2022年12月24日10:57:20
import torch
import torchvision
print(torch.cuda.is_available())

上面的命令只是检测CUDA是否安装正确并能被Pytorch检测到，并没有说明是否能正常使用，要想看Pytorch能不能调用cuda加速，还需要简单的测试一下：

a = torch.Tensor(5,3)
a = a.cuda()
print(a)

Cuda runtime error (48) : no kernel image is available for execution

处理办法：
一般来讲，CUDA的48号错误，是在于硬件的支持情况。对于算力3.0的显卡来说，如果安装了9.0的CUDA就会出现这个问题，解决的办法是退回CUDA8.0，或者更换更加高端的显卡，或者直接从源码编译，并在源码中做相应设置（修改setup.py文件里的TORCH_CUDA_ARCH_LIST，将这个值改成你当前使用的GPU对应算力！）

  • 作者：Axiiiz
  • 原文链接：https://blog.csdn.net/sinat_38059712/article/details/118162620
    更新时间：2022年12月24日10:57:20 ，共 405 字。