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