我正在使用两张图形卡来测试CUDA中的某些程序。我认为其中一个已损坏,但计算机仍然可以识别它。如何切换“主图形卡”以运行程序(如果存在)并发现它们是否正常运行?
答案 0 :(得分:1)
This可能会有所帮助。您可以使用cudaSetDevice
选择要在哪个GPU上运行。您也可以从命令行运行nvidia-smi
来检查哪些设备在系统上可见。请注意,nvidia-smi的顺序是PCIe顺序,而驱动程序和运行时并不保证这样做。
您可以使用cudaGetDeviceCount
来获取可见设备的总数,然后循环一次,使用循环索引作为参数调用cudaSetDevice
来设置要在其上运行的新设备
希望这会有所帮助!