Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix fairscale compatible with PT 1.8 (#5996) 

* try to extend fairscale available

* 1.2
This commit is contained in:
Jirka Borovec 2021-02-16 20:43:02 +01:00 committed by GitHub
parent c5919fde63
commit 960a60743f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/advanced/multi_gpu.rst
View File

@ -685,7 +685,7 @@ To get started, install FairScale using the command below. We install a specific
.. code-block:: bash
pip install https://github.com/PyTorchLightning/fairscale/archive/pl_1.1.0.zip
pip install https://github.com/PyTorchLightning/fairscale/archive/pl_1.2.0.zip
To use Sequential Model Parallelism, you must define a :class:`nn.Sequential <torch.nn.Sequential>` module that defines the layers you wish to parallelize across GPUs.
This should be kept within the ``sequential_module`` variable within your ``LightningModule`` like below.

2
requirements/extra.txt
View File

@ -7,4 +7,4 @@ torchtext>=0.5, <0.7 # TODO: temporary fix fix for compatibility
onnx>=1.7.0
onnxruntime>=1.3.0
hydra-core>=1.0
https://github.com/PyTorchLightning/fairscale/archive/pl_1.1.0.zip
https://github.com/PyTorchLightning/fairscale/archive/pl_1.2.0.zip