我想确定系统上的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