How to run NeuralProphet on GPU?

[frankelau]

Prerequisites

• Put an X between the brackets on this line if you have done all of the following:
  • Checked the Answered Questions on the Github Discussion board: https://github.com/ourownstory/neural_prophet/discussions
    If you have the same question but the Answer does not solve your issue, please continue the conversation there.
  • Checked that your issue isn't already filed: https://github.com/ourownstory/neural_prophet/issues
    If you have the same issue but there is a twist to your situation, please add an explanation there.
  • Considered whether your issue might need further discussing before being defined as a feature request:
    Please post an idea or feedback

Is your feature request related to a problem? Please describe.

A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]

Describe the solution you'd like

A clear and concise description of what you want to happen.

Describe alternatives you've considered

A clear and concise description of any alternative solutions or features you've considered.

Additional context

Add any other context or screenshots about the feature request here.

[quant-exchange]

Make sure you have this code run before you create a NeuralProphet instance (assuming you're using CUDA/NVIDIA GPU)*.

import torch

# Check if CUDA is available, if not fall back to CPU
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

# If CUDA is available, print information about each available GPU
if torch.cuda.is_available():
    for i in range(torch.cuda.device_count()):
        print(f"Device {i}: {torch.cuda.get_device_name(i)} (ID: {torch.cuda.get_device_properties(i)})")

    # Set the GPU device to use
    gpu_index = 0
    torch.cuda.set_device(gpu_index)
    print(f"Using device {gpu_index}: {torch.cuda.get_device_name(gpu_index)}")
else:
    # Fallback to CPU
    print("CUDA not available, using CPU.")