MLOps-Basics
MLOps-Basics copied to clipboard
gpus depected in new version of pytorchLightning
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