Skip to content

Commit

Permalink
temporary fix for cuda attempt 1
Browse files Browse the repository at this point in the history
  • Loading branch information
Ionshiv committed Sep 17, 2024
1 parent a0071a6 commit e264e21
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 3 deletions.
2 changes: 2 additions & 0 deletions .github/workflows/build-and-push-dockerimage.yml
Original file line number Diff line number Diff line change
Expand Up @@ -20,11 +20,13 @@ jobs:
tensorflow_version="${git_branch#tf-}" # remove prefix
tensorflow_version="${tensorflow_version%-*}" # remove suffix
base_image="tensorflow/tensorflow:${tensorflow_version}-gpu-jupyter"
framework="cuda"
image_name="pharmbio-notebook"
echo "IMAGE_NAME=$image_name" >> $GITHUB_ENV
echo "TAG=$tag" >> $GITHUB_ENV
echo "BASE_IMAGE=$base_image" >> $GITHUB_ENV
echo "FRAMEWORK=$framework" >> $GITHUB_ENV
- name: Set up Docker Buildx
uses: docker/setup-buildx-action@v1
Expand Down
2 changes: 2 additions & 0 deletions .inputrc
Original file line number Diff line number Diff line change
Expand Up @@ -83,3 +83,5 @@ set enable-bracketed-paste
set enable-bracketed-paste
set enable-bracketed-paste
set enable-bracketed-paste
set enable-bracketed-paste
set enable-bracketed-paste
4 changes: 2 additions & 2 deletions build-local-docker.sh
Original file line number Diff line number Diff line change
Expand Up @@ -11,11 +11,11 @@ echo "tag=$tag"
echo "image=$image"
#switch to buildx build
# Build docker image for this container.
docker buildx build -t ghcr.io/pharmbio/$image:$tag-devel \
docker buildx build --no-cache -t ghcr.io/pharmbio/$image:$tag-devel \
--build-arg BASE_IMAGE=tensorflow/tensorflow:${tensorflow_version}-jupyter \
--build-arg FRAMEWORK=cpu \
-f ./docker/env.Dockerfile . || exit 1
docker buildx build -t ghcr.io/pharmbio/$image:${tag}-gpu-devel \
docker buildx build --no-cache -t ghcr.io/pharmbio/$image:${tag}-gpu-devel \
--build-arg BASE_IMAGE=tensorflow/tensorflow:${tensorflow_version}-gpu-jupyter \
--build-arg FRAMEWORK=cuda \
-f docker/env.Dockerfile . || exit 1
Expand Down
2 changes: 1 addition & 1 deletion docker/env.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ ARG BASE_IMAGE="tensorflow/tensorflow:latest" # Example default
FROM $BASE_IMAGE

# ARG FRAMEWORK need to be after the build stage FROM (since it is used in another build process)
ARG FRAMEWORK="cpu"
ARG FRAMEWORK

# Set shell
ENV SHELL=/bin/bash
Expand Down

0 comments on commit e264e21

Please sign in to comment.