Skip to content

Commit

Permalink
working version, I hope
Browse files Browse the repository at this point in the history
  • Loading branch information
Ionshiv committed Oct 11, 2024
1 parent 3d72bee commit 0c83d36
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 12 deletions.
1 change: 1 addition & 0 deletions .inputrc
Original file line number Diff line number Diff line change
Expand Up @@ -103,3 +103,4 @@ set enable-bracketed-paste
set enable-bracketed-paste
set enable-bracketed-paste
set enable-bracketed-paste
set enable-bracketed-paste
2 changes: 1 addition & 1 deletion build-local-docker.sh
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ echo "image=$image"
# docker buildx build --no-cache -t ghcr.io/pharmbio/$image:$tag \
# --build-arg BASE_IMAGE=tensorflow/tensorflow:${tensorflow_version}-jupyter \
# -f ./docker/env.cpu.Dockerfile . || exit 1
DOCKER_BUILDKIT=1 docker buildx build --no-cache -t ghcr.io/pharmbio/$image:${tag}-gpu \
docker buildx build --no-cache -t ghcr.io/pharmbio/$image:${tag}-gpu \
--build-arg BASE_IMAGE=tensorflow/tensorflow:${tensorflow_version}-gpu-jupyter \
-f docker/env.cuda.Dockerfile . || exit 1

Expand Down
20 changes: 9 additions & 11 deletions docker/env.cuda.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,8 @@ RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
ENV DEBIAN_FRONTEND=noninteractive
# >apt_installs.txt to save instead of executing
RUN <<EOF
apt-get update
apt-get install -y --no-install-recommends \
apt-get update
apt-get install -y --no-install-recommends \
apt-transport-https \
ca-certificates \
software-properties-common \
Expand Down Expand Up @@ -61,18 +61,16 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip && \


# Breakpoint - temporary shell for debugging
RUN --mount=type=cache,target=/tmp \
echo "Entering breakpoint shell"; \
sleep infinity
# RUN --mount=type=cache,target=/tmp \
# echo "Entering breakpoint shell"; \
# sleep infinity

RUN echo "Step 2: Continuing after breakpoint"
# RUN echo "Step 2: Continuing after breakpoint"

RUN <<EOF
echo "Installing for CUDA framework"
python3 -m pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cu121 \
torch==2.4.1 \
torchvision \
torchaudio;
python3 -m pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cu121 torch==2.4.1 \
torchvision \
torchaudio;
EOF

# RUN python3 -m pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cu121 \
Expand Down

0 comments on commit 0c83d36

Please sign in to comment.