Having a hard time getting GPU in google colab to work with NeuralForecast()

[Avidar123]

I am currently experimenting and seeing what NeuralForecast has to offer, I am fairly new to this so please forgive me if this might seem obvious to some of you.

When I run the normal CPU in google colab NeuralForecast seems to work fine, but when i change the runtype to the T4 GPU it keeps giving me errors and saying that "ray.exceptions.ActorDiedError: The actor died unexpectedly before finishing this task" the full error will posted below with the code.

This is what it says at the bottom of the last image:
"The actor is dead because its worker process has died. Worker exit type: SYSTEM_ERROR Worker exit detail: Worker unexpectedly exits with a connection error code 2. End of file. There are some potential root causes. (1) The process is killed by SIGKILL by OOM killer due to high memory usage. (2) ray stop --force is called. (3) The worker is crashed unexpectedly due to SIGSEGV or other unexpected errors.
(raylet) A worker died or was killed while executing a task by an unexpected system error. To troubleshoot the problem, check the logs for the dead worker. RayTask ID: ffffffffffffffff0f571abf71432a2d3944e42a01000000 Worker ID: b35d77616f18d3e18ff11ed850855bf212f8fbe3490f4c95fd9c5eae Node ID: d0eb3994395a762aae5e0fa69c864d575624e1cc0391a89c6264485d Worker IP address: 172.28.0.12 Worker port: 43889 Worker PID: 9414 Worker exit type: SYSTEM_ERROR Worker exit detail: Worker unexpectedly exits with a connection error code 2. End of file. There are some potential root causes. (1) The process is killed by SIGKILL by OOM killer due to high memory usage. (2) ray stop --force is called. (3) The worker is crashed unexpectedly due to SIGSEGV or other unexpected errors."

[Avidar123]

Found a solution for this, i just moved my code to an entirely new colab file and it worked perfectly. must've done something to the colab settings to make it act this way