如何确定系统上的CUDA设备的数量(而无需编译任何内容)



我想确定系统上的CUDA设备数量。当然,我可以使用C程序中的cuda运行时API来进行此操作,但是 - 没有编译任何代码的情况下,最简单/最简单的方法是执行此操作的最简单/最直接的方法?

假设:

  • cuda> = 7.0已安装
  • 最新的NVIDIA驱动程序已安装
  • 您可以为Unix状和非unix样OS提供不同的解决方案。

如果您在Linux或类似Unix的系统上,这有效:

nvidia-smi  -L | wc -l

说明:

  • nvidia-smi是nvidia的"系统管理接口"公用事业。您可以使用它来确定有关GPU的各种信息,并执行一些维护操作。
  • nvidia-smi -L列出了系统上可用的GPU。
  • wc是unix word/line/conture Counting实用程序
  • wc -l计算标准输入上的行数

...但是有点脆弱。我希望我有一些更健壮的东西。

这也有效:

(iit_synthesis) [miranda9@cccxc432 ~]$ echo $CUDA_VISIBLE_DEVICES
0,1

相关内容

最新更新