如何知道我的程序在哪个图形卡上运行?

时间:2019-04-05 19:30:55

标签: cuda nvidia

我正在使用两张图形卡来测试CUDA中的某些程序。我认为其中一个已损坏,但计算机仍然可以识别它。如何切换“主图形卡”以运行程序(如果存在)并发现它们是否正常运行?

1 个答案:

答案 0 :(得分:1)

This可能会有所帮助。您可以使用cudaSetDevice选择要在哪个GPU上运行。您也可以从命令行运行nvidia-smi来检查哪些设备在系统上可见。请注意,nvidia-smi的顺序是PCIe顺序,而驱动程序和运行时并不保证这样做。

您可以使用cudaGetDeviceCount来获取可见设备的总数,然后循环一次,使用循环索引作为参数调用cudaSetDevice来设置要在其上运行的新设备

希望这会有所帮助!