mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-08-22 12:20:50 +08:00
William Falcon
4c6c3d04ce
* added outline of all features * updated common use cases doc * updated common use cases doc * updated common use cases doc * updated common use cases doc * updated common use cases doc * updated common use cases doc * updated common use cases doc * updated common use cases doc * updated common use cases doc * updated common use cases doc * updated common use cases doc * updated common use cases doc * updated common use cases doc * updated docs
9 lines
287 B
ReStructuredText
9 lines
287 B
ReStructuredText
Single GPU Training
|
|
====================
|
|
Make sure you are running on a machine that has at least one GPU. Lightning handles all the NVIDIA flags for you,
|
|
there's no need to set them yourself.
|
|
|
|
.. code-block:: python
|
|
|
|
# train on 1 GPU (using dp mode)
|
|
trainer = pl.Trainer(gpus=1) |