在当前机器上获取 sm 版本的 gpu 命令
Command to get sm version of gpu in current machine
是否有获取给定机器中 sm 版本的 gpu 的命令。这是我的用例：我在多台机器上构建和 运行 相同的 cuda 内核。所以我想知道是否有一个命令可以检测给定系统上 gpu 的 sm 版本并将其作为参数传递给 nvcc:
$ nvcc -arch=`gpuarch -device 0` mykernel.cu
我找到了 https://gist.github.com/f0k/63a664160d016a491b2cbea15913d549。我发现这是最好的解决方案。你只需要 python.
是否有获取给定机器中 sm 版本的 gpu 的命令。这是我的用例：我在多台机器上构建和 运行 相同的 cuda 内核。所以我想知道是否有一个命令可以检测给定系统上 gpu 的 sm 版本并将其作为参数传递给 nvcc:
$ nvcc -arch=`gpuarch -device 0` mykernel.cu
我找到了 https://gist.github.com/f0k/63a664160d016a491b2cbea15913d549。我发现这是最好的解决方案。你只需要 python.