问题很简单：我的一个用户在pytorch中遇到了一个错误：
Found GPU0 NVIDIA GeForce GT 710 which is of cuda capability 3.5.
PyTorch no longer supports this GPU because it is too old.
The minimum cuda capability supported by this library is 3.7.这意味着我不能依赖torch.cuda.is_available()来检查它是否真的可以运行我的代码。我还需要确保GPU的CUDA计算能力〉= 3.7。然而，出于某种奇怪的原因，在堆栈溢出或谷歌上找不到获得GPU CUDA计算能力这一看似简单的任务，尽管错误消息中有这一点，但它似乎没有问题。
1条答案
按热度按时间yhived7q1#
我没有意识到的是，
torch.cuda.get_device_properties(0)的“主要”和“次要”实际上是CUDA计算能力，例如，如果主要是7，次要是5，则CUDA能力是7.5，也可以通过get_device_capability来完成。