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.