Skip to content

Commit

Permalink
Fine-tune [Skip CI]
Browse files Browse the repository at this point in the history
  • Loading branch information
YanWenKun committed Apr 11, 2024
1 parent 5f8e4d6 commit 9536238
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 10 deletions.
16 changes: 8 additions & 8 deletions megapak/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,7 @@ FROM opensuse/tumbleweed:latest

LABEL maintainer="code@yanwk.fun"

WORKDIR /root

RUN set -euxo pipefail
RUN set -eu

################################################################################
# NVIDIA CUDA devel
Expand Down Expand Up @@ -43,17 +41,16 @@ cuda-nvtx-12-1 \
libcublas-12-1 \
libcublas-devel-12-1 \
libnpp-12-1 \
libnpp-devel-12-1 \
&& env
libnpp-devel-12-1

ENV PATH="${PATH}:/usr/local/cuda-12.1/bin" \
LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/usr/local/cuda-12.1/lib64" \
LIBRARY_PATH="${LIBRARY_PATH}:/usr/local/cuda-12.1/lib64/stubs" \
CUDA_HOME="/usr/local/cuda-12.1"

################################################################################
# Python, GCC and tools
# Since this image is so big, here we use openSUSE-verified PIP packages for compatibility.
# Python and tools
# Since this image is so big, we use openSUSE-verified PIP packages for compatibility.

RUN --mount=type=cache,target=/var/cache/zypp \
zypper addrepo --check --refresh --priority 90 \
Expand Down Expand Up @@ -108,7 +105,10 @@ google-noto-sans-cjk-fonts \
google-noto-coloremoji-fonts \
&& rm /usr/lib64/python3.11/EXTERNALLY-MANAGED

# Fix for GCC12 (required for compiling CUDA 12.1-related code)
################################################################################
# GCC 12
# Required for compiling CUDA 12.1-related code.

RUN --mount=type=cache,target=/var/cache/zypp \
zypper --gpg-auto-import-keys \
install --no-confirm --auto-agree-with-licenses \
Expand Down
3 changes: 1 addition & 2 deletions rocm/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -84,8 +84,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
mediapipe \
&& pip list

RUN env \
&& du -ah /root \
RUN du -ah /root \
&& rm -rf /root/* /root/.*

COPY runner-scripts/. /runner-scripts/
Expand Down

0 comments on commit 9536238

Please sign in to comment.