命令在当前机器中获取GPU的SM版本

Question

是否有命令在给定的计算机中获取GPU的SM版本。这是我的用例：我在多台计算机上构建并运行相同的CUDA内核。因此，我想知道是否有一个命令可以在给定系统上检测SM版本的SM版本，并将其作为争论传递给NVCC：

$ nvcc -arch=`gpuarch -device 0` mykernel.cu

Answer 1

我找到了https://gist.github.com/f0k/63a664160d016a491b2cbea15913d549。我发现这是最好的解决方案。您只需要python。