Cuda

Question

Cuda

Opened this issue 3 years ago · 3 comments

Bturan19 commented 3 years ago

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

Answer 1 · 2021-09-09T07:33:38.000Z

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)

Answer 2 · 2021-09-09T09:01:38.000Z

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

Answer 3 · 2021-09-13T06:59:25.000Z

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.