mxnet.base.MXNetError: [10:32:39] src/engine/./…/common/cuda_utils.h:395: Check failed: e == cudaSuccess || e == cudaErrorCudartUnloading: CUDA: invalid device ordinal
上述错误,说明当前使用cuda的编号出现错误
解决方法:
仔细查看gpu的id。有可能os.environ[‘CUDA_VISIBLE_DEVICES’] 设置为一张卡,但是gpu的编号取到1。