Skip to content

Try to reduce torch img size #127

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 11 commits into from
Sep 22, 2025
Merged

Try to reduce torch img size #127

Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
ff7c016
try to reduce torch img size
Sep 22, 2025
f09f5ef
typo
Sep 22, 2025
3f45840
debug
Sep 22, 2025
4e60bc5
debug cuda
Sep 22, 2025
1153e6a
debug
Sep 22, 2025
362d93a
debug apt install
Sep 22, 2025
0637374
debug apt-get install
Sep 22, 2025
dcfe00b
debug apt install
Sep 22, 2025
6b624d6
apt-get update
Sep 22, 2025
65adc5c
fix for non-torch situation
Sep 22, 2025
d49edf3
fix for paddle img size
Sep 22, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 5 additions & 2 deletions docker_core/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -83,8 +83,6 @@ RUN set -eux \
&& ( $(grep -q "paddle" <<< "${ARG_PROFILE_PYTHON}") && ( \
URL_PYPI_PADDLE="https://www.paddlepaddle.org.cn/packages/stable/${IDX}/" \
&& PADDLE=$( [ -x "$(command -v nvcc)" ] && echo "paddlepaddle-gpu" || echo "paddlepaddle") \
# && PADDLE_VER=$(pip index versions ${PADDLE} -f ${URL_PYPI_PADDLE} | grep 'Available' | cut -d ":" -f 2 | tr ', ' '\n' | grep ${CUDA_VER:-'.'} | head -n 1) \
# && V=$(echo ${PADDLE}==${PADDLE_VER}) && echo "to install paddle: ${V}" \
&& pip install --no-cache-dir --root-user-action=ignore -U --pre --index-url ${URL_PYPI_PADDLE} "${PADDLE}" \
) || echo "Skip paddle install" ) \
# -----------------------------
Expand All @@ -93,6 +91,11 @@ RUN set -eux \
[ -f "/opt/utils/install_list_PY_${profile}.apt" ] && install_apt "/opt/utils/install_list_PY_${profile}.apt" || echo "apt install skipped for ${profile}" ; \
[ -f "/opt/utils/install_list_PY_${profile}.pip" ] && install_pip "/opt/utils/install_list_PY_${profile}.pip" || echo "pip install skipped for ${profile}" ; \
) ; done \
&& if [ "$(echo "${IDX}" | cut -c1-2)" = "cu" ] && echo "${ARG_PROFILE_PYTHON}" | grep -qE "torch|paddle" ; then \
echo "Try to uninstall nvidia python packages to reduce storage size..." \
&& pip freeze | grep -i '^nvidia-' | cut -d'=' -f1 | xargs -r pip uninstall -y \
&& apt-get -qq update --fix-missing && apt-get -qq install -y --no-install-recommends --allow-change-held-packages libcusparselt0 libnccl2 libnccl-dev ; \
fi \
# -----------------------------
&& if echo "${ARG_PROFILE_GO}" | grep -q "base" ; then \
echo "Installing GO: ${ARG_PROFILE_GO}" && setup_GO ; \
Expand Down