在当前机器上获取 sm 版本 gpu 的命令

Question

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

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

Answer 1

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