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 (#7787)

This commit is contained in:
Jirka Borovec 2021-06-01 12:11:29 +02:00 committed by GitHub
parent 0b6fd1da54
commit 9a001fea22
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
dockers/nvidia/Dockerfile
View File

@ -13,7 +13,7 @@
# limitations under the License.
# https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes
FROM nvcr.io/nvidia/pytorch:21.04-py3
FROM nvcr.io/nvidia/pytorch:21.05-py3
LABEL maintainer="PyTorchLightning <https://github.com/PyTorchLightning>"
@ -39,14 +39,15 @@ RUN \
# Installations
python -c "fname = './pytorch-lightning/requirements/extra.txt' ; lines = [line for line in open(fname).readlines() if not line.startswith('horovod')] ; open(fname, 'w').writelines(lines)" && \
pip install "Pillow>=8.2" "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.1" --no-cache-dir --upgrade-strategy only-if-needed && \
rm -rf pytorch-lightning && \
pip list
RUN pip install lightning-grid -U
RUN pip install lightning-grid -U && \
pip install "py>=1.10" --upgrade-strategy only-if-needed
ENV PYTHONPATH="/workspace"