我想用gpu训练模型，但是当我给出gpu的索引时，总会出现“ src / storage / storage.cc：65：检查失败：e == cudaSuccess || e == cudaErrorCudartUnloading CUDA：无效的设备序号“
我已经用“ nvidia-smi”检查了gpu的数量，我的服务器上只有2个gpu，编号为“ 0”和“ 1”。而且我用过mxnet。
os.environ [“ CUDA_DEVICE_ORDER”] =“ PCI_BUS_ID” os.environ [“ CUDA_VISIBLE_DEVICES”] =“ 0,1”