Cuda

[Bturan19]

Thanks for this awesome project firstly.

I wonder if its possible to fit a model on GPU.

In predict method I see that there is a cuda parameter but couldnt find it in fit parameter.

Thanks in advance

[KIC]

I have never really tested cuda since I don't have an nvidia any more. so without any guarantee it should be good enough to just pass cuda=True to the kwargs of your fit function.

with df.model() as m:
  m.fit(....., cuda=True)

[Bturan19]

Thanks for answer. It didn't work just like that because all the data was in cpu. Some additional work is needed there I guess

[KIC]

Since I don't have a cuda gpu at the moment you might try to fix it yourself and submit a PR. It should not be too complicated as there are not many places for data conversion. it is utils.from_pandas or utils.to_device which might miss the flag.