Which Nvidia GPUs does PyTorch support?

1

I’m trying to find a list of which Nvidia GPUs are supported by which PyTorch versions. Seems like this should be in the PyTorch documentation somewhere.

Specifically, I’m getting this error:
“CUDA error: no kernel image is available for execution on the device”

Wondering if the version I’m using just doesn’t support my GPU any more.

2

You can check the supported compute capabilities via torch.cuda.get_arch_list() and check if your GPU is too old.