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...

score 1 · Answer 1 · answered Oct 25 '21 at 12:20

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

answered Oct 25 '21 at 12:20

Saravanan Natarajan

355
1
6

Yes you are right I am not using GPU version any idea how to install it? – Sharan Oct 25 '21 at 12:21
first remove your current mxnet version then install pip install mxnet-cu92 for linux system. Please find https://mxnet.apache.org/versions/1.4.1/install/index.html?platform=Linux&language=Python&processor=GPU – Saravanan Natarajan Oct 25 '21 at 12:47

Using DeepAR GPU

1 Answers1