Merge branch 'master' into FSDP

This commit is contained in:
Luca Antiga 2024-12-10 17:56:51 +01:00 committed by GitHub
commit e15ee2d6d1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -39,7 +39,7 @@ RUN \
fi && \ fi && \
# otherwise there is collision with folder name and pkg name on Pypi # otherwise there is collision with folder name and pkg name on Pypi
cd pytorch-lightning && \ cd pytorch-lightning && \
pip install setuptools && \ pip install setuptools==75.6.0 && \
PACKAGE_NAME=lightning pip install '.[extra,loggers,strategies]' --no-cache-dir && \ PACKAGE_NAME=lightning pip install '.[extra,loggers,strategies]' --no-cache-dir && \
PACKAGE_NAME=pytorch pip install '.[extra,loggers,strategies]' --no-cache-dir && \ PACKAGE_NAME=pytorch pip install '.[extra,loggers,strategies]' --no-cache-dir && \
cd .. && \ cd .. && \