PyTorch get_device_capability()输出解释

Question

在我的GTX 1070上运行torch.cuda.get_device_capability()时，我得到以下输出：(6, 1)。

有人能解释一下这意味着什么吗？

Answer 1

正如注释中所述，这些实际上是GPU的计算能力版本/指示器。简单地说，它描述了GPU支持的特性。如果您在windows上，也可以使用GPU查看您的GPU计算能力。

有关确切的特性集差异的更多信息，您可以查看这里。

正如您在这里所看到的，基于这些，开发人员可以知道哪些功能集可供开发人员使用，因此，如果他们支持某些硬件，并以其他方式支持它们的实现，则可以为某些硬件启用一些特性。

这可能也很有用。