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

Merge branch 'master' into FSDP

This commit is contained in:
Luca Antiga 2024-12-10 15:42:16 +01:00 committed by GitHub
parent c92fea8d6f b7099851b7
commit d3734e6f14
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
dockers/base-cuda/Dockerfile
View File

@ -59,7 +59,6 @@ RUN \
add-apt-repository ppa:deadsnakes/ppa && \ add-apt-repository ppa:deadsnakes/ppa && \
apt-get install -y \ apt-get install -y \
python${PYTHON_VERSION} \ python${PYTHON_VERSION} \
python3-setuptools \
python${PYTHON_VERSION}-dev \ python${PYTHON_VERSION}-dev \
&& \ && \
update-alternatives --install /usr/bin/python${PYTHON_VERSION%%.*} python${PYTHON_VERSION%%.*} /usr/bin/python${PYTHON_VERSION} 1 && \ update-alternatives --install /usr/bin/python${PYTHON_VERSION%%.*} python${PYTHON_VERSION%%.*} /usr/bin/python${PYTHON_VERSION} 1 && \
@ -79,6 +78,8 @@ RUN \
curl https://bootstrap.pypa.io/get-pip.py | python${PYTHON_VERSION} && \ curl https://bootstrap.pypa.io/get-pip.py | python${PYTHON_VERSION} && \
# Disable cache \ # Disable cache \
pip config set global.cache-dir false && \ pip config set global.cache-dir false && \
# Install recent setuptools to obtain pkg_resources \
pip install setuptools==75.6.0 && \
# set particular PyTorch version \ # set particular PyTorch version \
pip install -q wget packaging && \ pip install -q wget packaging && \
python -m wget https://raw.githubusercontent.com/Lightning-AI/utilities/main/scripts/adjust-torch-versions.py && \ python -m wget https://raw.githubusercontent.com/Lightning-AI/utilities/main/scripts/adjust-torch-versions.py && \