【发布时间】:2019-04-05 19:30:54
【问题描述】:
我正在使用两张显卡来测试 CUDA 中的一些程序。我认为其中一个坏了,但计算机仍然可以识别它。如何切换“主显卡”以运行我的程序(如果存在)并发现它们是否正常工作?
【问题讨论】:
我正在使用两张显卡来测试 CUDA 中的一些程序。我认为其中一个坏了,但计算机仍然可以识别它。如何切换“主显卡”以运行我的程序(如果存在)并发现它们是否正常工作?
【问题讨论】:
This 可能会有所帮助。您可以使用cudaSetDevice 选择要在哪个 GPU 上运行。您还可以从命令行运行nvidia-smi 来检查您的系统上可以看到哪些设备。请注意,nvidia-smi 的顺序是 PCIe 顺序,而驱动程序和运行时不保证这一点。
您可以使用cudaGetDeviceCount 获取可见设备的总数,然后循环该次数,调用cudaSetDevice 并将循环的索引作为参数来设置要在其上运行的新设备。
希望这会有所帮助!
【讨论】: