0

I tried to run Imputer.fit by passing mxnet.context with gpu id but it seems mycode still run on cpu.

from mxnet import gpu
...
ctx = gpu(0)
...
imputer.fit(train_df=imputer_train_df, num_epochs=10, ctx=ctx)

Anyone know how to correctly do this ? I understand that mxnet doesn't support cuda 11.0 that I use now.

Is there a way to run datawig on GPU ?

talonmies
  • 70,661
  • 34
  • 192
  • 269
3ORZ
  • 79
  • 1
  • 3

1 Answers1

0

Do you get an error/warning message? GPU support is MXNet based, so having a compatible versions should be a requirement.

tammo
  • 21
  • 1
  • 5