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