MLOps-Basics icon indicating copy to clipboard operation
MLOps-Basics copied to clipboard

gpus depected in new version of pytorchLightning

Open Sauravdas3 opened this issue 5 months ago • 0 comments

in the PL.Trainer ( # the updated code should be===> # Use "gpu" if available, else "cpu" accelerator="gpu" if torch.cuda.is_available() else "cpu",

                 devices=1,        # Always use 1 device, whether it's GPU or CPU

Sauravdas3 avatar Sep 29 '24 19:09 Sauravdas3