Using mxnet with CUDA freezes python

Performance
1

I tried to install mxnet with CUDA support and eventually succeeded. Although the installation finally worked, all my python applications are freezing whenever I try to actually use it. To be more precise, I’m trying to run the validation code:

import mxnet as mx
a = mx.nd.ones((2, 3), mx.gpu())
b = a * 2 + 1
b.asnumpy()

It runs fine if I leave out the gpu part.

My CUDA version is 10.2.89 and cuDNN is version 7.6.5. My GPU is the 3060Ti.

2

Did you run it on Windows or Linux? What is the output of the nvidia-smi command if you run it on Linux?