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

update NGC docker (#13136) 

* update docker
* Apply suggestions from code review

Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
This commit is contained in:
Jirka Borovec 2022-06-02 14:54:13 +02:00 committed by GitHub
parent 4819ad1c58
commit de4ab1c027
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
dockers/nvidia/Dockerfile
View File

@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
ARG PYTORCH_VERSION=21.11
ARG PYTORCH_VERSION=22.04
# https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes
FROM nvcr.io/nvidia/pytorch:${PYTORCH_VERSION}-py3
@ -37,20 +37,19 @@ RUN \
cd .. ; \
fi && \
# save the examples
mv pytorch-lightning/_notebooks notebooks && \
mv pytorch-lightning/_notebooks/.notebooks/ notebooks && \
mv pytorch-lightning/pl_examples . && \
# Installations \
pip install -q fire && \
python ./pytorch-lightning/.actions/assistant.py requirements_prune_pkgs horovod --req_files ./pytorch-lightning/requirements/extra.txt && \
pip install "Pillow>=8.2, !=8.3.0" "cryptography>=3.4" "py>=1.10" --no-cache-dir --upgrade-strategy only-if-needed && \
pip install -r ./pytorch-lightning/requirements/extra.txt --no-cache-dir --upgrade-strategy only-if-needed && \
pip install -r ./pytorch-lightning/requirements/examples.txt --no-cache-dir --upgrade-strategy only-if-needed && \
pip install ./pytorch-lightning --no-cache-dir && \
pip install "Pillow>=8.2, !=8.3.0" "cryptography>=3.4" "py>=1.10" --no-cache-dir && \
pip install ./pytorch-lightning["extra","loggers","strategies"] --no-cache-dir && \
pip install -r ./pytorch-lightning/requirements/examples.txt --no-cache-dir && \
rm -rf pytorch-lightning && \
pip install jupyterlab[all] -U && \
pip list
RUN pip install jupyterlab[all] -U
RUN pip install lightning-grid -U && \
pip install "py>=1.10" "protobuf>=3.15.6" --upgrade-strategy only-if-needed

2
dockers/release/Dockerfile
View File

@ -36,7 +36,7 @@ RUN \
mv pytorch-lightning-*/ pytorch-lightning ; \
rm *.zip ; \
fi && \
pip install ./pytorch-lightning["extra"] --no-cache-dir && \
pip install ./pytorch-lightning["extra","loggers","strategies"] --no-cache-dir && \
rm -rf pytorch-lightning
RUN python --version && \