Using DeepAR GPU

Question

I am using Gluonts for building DeepAR model but takes lot of time to run the training object eventhough I use cox = 'gpu' but throws an error. My machine has GPU but the option didn't work. Any help is much appreciated...

Answer 1

You can check your mxnet current version, I believe ur using a CPU version.

please check the following:

import mxnet as mx
    print(f'mxnet version: {mx.__version__}')
    print(f'Number of GPUs: {mx.context.num_gpus()}')

it should return number of gpus