Closed atward424 closed 6 years ago
In the config.mk file, you should modify the line to be "USE_CUDA=1" before compiling.
Please refer to this message:
src/storage/storage.cc:113: Compile with USE_CUDA=1 to enable GPU usage
It means MXNet is not compiled correctly so it cannot access the GPUs.
You can go to mxnet/make/config.mk
to modify the configs as follows:
USE_CUDA = 1
USE_CUDA_PATH = /usr/local/cuda
USE_CUDNN = 1
README.md is also updated.
That did it, thanks!
Hi,
I'm trying to run the Cityscapes classifier in test mode on an Amazon EC2 instance, but I'm running into this error:
Any idea what is causing it and how to fix it?
System: Ubuntu 16.04.3 mxnet 0.11.0 numpy 1.13.3 cv2 3.2.0 PIL 1.1.7 cython 0.27.1
Thanks!