在当前机器

Question

是否有命令在给定的机器中获取gpu的sm版本。这是我的用例：我在多台机器上构建并运行相同的cuda内核。所以我想知道是否有一个命令可以检测给定系统上的gpu版本，并将其作为争论传递给nvcc：

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

Answer 1

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