Merge branch 'master' into FSDP
This commit is contained in:
commit
d3734e6f14
|
@ -59,7 +59,6 @@ RUN \
|
|||
add-apt-repository ppa:deadsnakes/ppa && \
|
||||
apt-get install -y \
|
||||
python${PYTHON_VERSION} \
|
||||
python3-setuptools \
|
||||
python${PYTHON_VERSION}-dev \
|
||||
&& \
|
||||
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} && \
|
||||
# Disable cache \
|
||||
pip config set global.cache-dir false && \
|
||||
# Install recent setuptools to obtain pkg_resources \
|
||||
pip install setuptools==75.6.0 && \
|
||||
# set particular PyTorch version \
|
||||
pip install -q wget packaging && \
|
||||
python -m wget https://raw.githubusercontent.com/Lightning-AI/utilities/main/scripts/adjust-torch-versions.py && \
|
||||
|
|
Loading…
Reference in New Issue