CUDAAccelerator¶
- class lightning.pytorch.accelerators.CUDAAccelerator[source]¶
Bases:
Accelerator
Accelerator for NVIDIA CUDA devices.
- setup(trainer)[source]¶
Called by the Trainer to set up the accelerator before the model starts running on the device.