python -c "import torch; print(torch.cuda.get_arch_list())"
Find the architecture for a GPU with the following command:

CUDA_VISIBLE_DEVICES=0 python -c "import torch; print(torch.cuda.get_device_capability())"

To find the architecture for GPU 0: