Skip to content

Commit d961a77

Browse files
committed
pytorch: update changelog and comment
1 parent 63f72e2 commit d961a77

2 files changed

Lines changed: 6 additions & 2 deletions

File tree

ML-Frameworks/pytorch-aarch64/CHANGELOG.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,9 @@ where `YY` is the year, and `MM` the month of the increment.
1818
- Delete unused submodules to avoid fetching.
1919
- Replace google-perftool bundle with specific libtcmalloc-minimal4 and add symlink so that you can still use it with `LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libtcmalloc.so.4`.
2020
- Remove unused git apt package.
21+
- Remove outdated versions of pip/setuptools/wheel from image after pip updates itself.
22+
- Remove manual install of transformers in `transformers_llm_text_gen.py`, we now use a single version everywhere.
23+
- Remove unused pip packages: datasets, omegaconf, psutil, tiktoken, torchaudio, torchdata and torchtune.
2124

2225
### Fixed
2326

ML-Frameworks/pytorch-aarch64/Dockerfile

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,8 +65,9 @@ RUN echo 'export PATH="$HOME/.local/bin:$PATH"' >> /etc/bash.bashrc
6565
WORKDIR /home/$DOCKER_USER
6666
USER $DOCKER_USER
6767

68-
# Update to newer pip/setuptools/wheel (setuptools>= 70.0.0 due to CVE-2024-6345 and CVE-2025-47273)
69-
# and delete old system version (we essentially use apt:python3-pip to bootstrap pip)
68+
# Update to newer pip/setuptools/wheel (setuptools>= 70.0.0 due to CVE-2024-6345
69+
# and CVE-2025-47273, wheel >= 0.38.0 due to CVE-2022-40898) and delete old system
70+
# version (we essentially use apt:python3-pip to bootstrap pip)
7071
RUN pip install --upgrade pip~=25.2 setuptools~=78.1.1 wheel~=0.45.1 \
7172
&& sudo rm -r /usr/lib/python3/dist-packages/
7273

0 commit comments

Comments
 (0)