Using mxnet with CUDA freezes python

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

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.

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