我正在使用一些假设系统上有多个 GPU 的软件。它崩溃了,并显示错误消息:
[09:39:34] src/storage/storage.cc:71: Check failed: e == cudaSuccess || e == cudaErrorCudartUnloading CUDA: invalid device ordinal
我只有一张 GPU 卡。有什么方法可以欺骗 CUDA,让它只使用这张卡,而不会失败吗?
我可以在源代码中进行更改以仅使用单个设备,但我无法将此类更改提交到存储库,因此目前需要在分支上进行所有测试,并定期提取更改。
我的操作系统是 4.18.0-18-generic #19~18.04.1-Ubuntu。