You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
There is an error shows: RuntimeError: Lightning can't create new processes if CUDA is already initialized. Did you manually call torch.cuda.* functions, have moved the model to the device, or allocated memory on the GPU any other way? Please remove any such calls, or change the selected strategy. You will have to restart the Python kernel.
The text was updated successfully, but these errors were encountered:
I have 4 gpus and trying to train TFT model by setting up pl_trainer_kwargs as follows
pl_trainer_kwargs = {
"accelerator": "gpu",
"strategy": "ddp_notebook",
"devices": 4,
"callbacks": callbacks
}
There is an error shows: RuntimeError: Lightning can't create new processes if CUDA is already initialized. Did you manually call
torch.cuda.*
functions, have moved the model to the device, or allocated memory on the GPU any other way? Please remove any such calls, or change the selected strategy. You will have to restart the Python kernel.The text was updated successfully, but these errors were encountered: