mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-10 12:21:57 +08:00
Docs (#813)
* 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
This commit is contained in:
@@ -0,0 +1,9 @@
|
||||
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)
|
||||
Reference in New Issue
Block a user