Skip to content

Commit

Permalink
Dockerfile autobuild
Browse files Browse the repository at this point in the history
  • Loading branch information
FlyingFathead committed Oct 13, 2024
1 parent b82bf43 commit 8ed8106
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ ENV DEBIAN_FRONTEND=noninteractive
ENV LANG=C.UTF-8

# Install build dependencies and CUDA libraries
RUN apt-get update && apt-get install -y --no-install-recommends \
RUN apt-get update && apt-get install --allow-change-held-packages -y --no-install-recommends \
wget \
build-essential \
gcc-10 g++-10 \
Expand Down Expand Up @@ -83,7 +83,7 @@ ENV DEBIAN_FRONTEND=noninteractive
ENV LANG=C.UTF-8

# Install runtime dependencies
RUN apt-get update && apt-get install -y --allow-change-held-packages --no-install-recommends \
RUN apt-get update && apt-get install --allow-change-held-packages -y --no-install-recommends \
libnvidia-ml-dev \
libcudnn9-cuda-12 \
libcudnn9-dev-cuda-12 \
Expand All @@ -97,7 +97,7 @@ RUN apt-get update && apt-get install -y --allow-change-held-packages --no-insta
&& rm -rf /var/lib/apt/lists/*

# Optional: List held packages for debugging
RUN apt-mark showhold
# RUN apt-mark showhold

# Upgrade pip and install runtime Python dependencies
RUN python3 -m pip install --upgrade pip
Expand Down

0 comments on commit 8ed8106

Please sign in to comment.