Open1

NVIDIA GPU の Compute Capability を調べるための1行コマンド (PyTorch使用)

PINTOPINTO
$ python3 -c "import torch;print(torch.cuda.get_device_name());capa=torch.cuda.get_device_capability();print(f'Compute Capability: {capa[0]}{capa[1]}')"
NVIDIA GeForce RTX 3070
Compute Capability: 86

NVIDIA GeForce GTX 1650
Compute Capability: 75