CUDAAccelerator¶
- class lightning_fabric.accelerators.CUDAAccelerator[source]¶
Bases:
lightning_fabric.accelerators.accelerator.Accelerator
Accelerator for NVIDIA CUDA devices.
- setup_device(device)[source]¶
- Raises
ValueError – If the selected device is not of type CUDA.
- Return type