GPUAccelerator¶
-
class
pytorch_lightning.accelerators.
GPUAccelerator
(precision_plugin, training_type_plugin)[source]¶ Bases:
pytorch_lightning.accelerators.accelerator.Accelerator
Accelerator for GPU devices.
- Parameters
-
setup
(trainer, model)[source]¶ - Raises
MisconfigurationException – If the selected device is not GPU.
- Return type