You would have to build PyTorch from source, as this compute capability is not shipped in the binaries anymore since PyTorch 1.5, if I’m not mistaken.
Contributions to improve these error messages are always welcome.