MPS (Metal Performance Shaders) support for training

Question

MPS (Metal Performance Shaders) support for training

ehsanfathi77 opened this issue 4 months ago · 3 comments

ehsanfathi77 commented 4 months ago

When I train a model on my Mac I get this warning:

MPS available but not used. Set accelerator and devices using Trainer(accelerator='mps', devices=1).

Is there a way to take advantage of MPS to accelerate the training?

Answer 1 · 2024-05-15T17:42:55.000Z

it seems this works:

trainer_configs = {
    "accelerator": "mps"}

m = NeuralProphet(
    **trainer_configs
)

Answer 2 · 2024-06-17T18:32:10.000Z

@ehsanfathi77 Thanks for reporting this issue. Could you provide the exact logs that you are getting and the macOS version of your machine please ?

Answer 3 · 2024-06-19T20:03:21.000Z

We have thoroughly investigated this issue, and we were unable to find or reproduce the reported problem. As a result, we will be closing this issue for now. If the issue persists, please feel free to reopen it.