How to run datawig on GPU?

前端 未结 0 1146
栀梦
栀梦 2020-12-29 22:40

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(t         


        
相关标签:
回答
  • 消灭零回复
提交回复
热议问题