0

I am running a neural net of gpu cloud platform with 4 GPUs A100.When I run the code for training I am getting this error:

MX call MXNDArrayCreateEx failed... MXNetError: Compile with USE_CUDA=1 to enable GPU usage

From the support suggested couple of solutions, none of which is helping, except to check the compatibility b/w CUDA and MXNet.So typing nvidia-smi shows CUDA version is 12.0, what is the corresponding version of MXNet for it, and how to install it?

I tried reducing the batch_size and checking the GPU's memory but none of them seems to change the error.

Aleksandar
  • 1

1 Answers1

0

there is currently no MXNet version supporting CUDA 12.0 as of now. Looks like the only option is downgrading CUDA or switching over to use the CPU.

For downgrading check out this tutorial:

https://www.youtube.com/watch?v=5eJTzhGe2QE

For using the CPU simply change from mx.gpu() to mx.cpu().

Hope they'll update soon and support CUDA 12.x versions.

SickerDude43
  • 168
  • 8