Try with raw push

huggingface · Aug 30, 2023 · bde97a3 · bde97a3
1 parent 6bed440
commit bde97a3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 14 deletions.
diff --git a/.github/workflows/build_docker_images.yml b/.github/workflows/build_docker_images.yml
@@ -24,14 +24,10 @@ jobs:
         with:
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_PASSWORD }}
-
       - name: Build and Push CPU
-        uses: docker/build-push-action@v2
-        with:
-          context: ./docker/accelerate-cpu
-          push: true
-          tags: huggingface/accelerate-cpu
-          driver: docker
+        run: |
+          docker build -t huggingface/accelerate-cpu -f ./docker/accelerate-cpu/Dockerfile .
+          docker push huggingface/accelerate-cpu:latest
 
   latest-cuda:
     name: "Latest Accelerate GPU [dev]"
@@ -48,9 +44,6 @@ jobs:
           password: ${{ secrets.DOCKERHUB_PASSWORD }}
 
       - name: Build and Push GPU
-        uses: docker/build-push-action@v2
-        with:
-          context: ./docker/accelerate-gpu
-          push: true
-          tags: huggingface/accelerate-gpu
-          driver: docker
+        run: |
+          docker build -t huggingface/accelerate-gpu -f ./docker/accelerate-gpu/Dockerfile .
+          docker push huggingface/accelerate-gpu:latest
diff --git a/docker/accelerate-gpu/Dockerfile b/docker/accelerate-gpu/Dockerfile
@@ -12,7 +12,7 @@ RUN apt-get update && \
     rm -rf /var/lib/apt/lists*
 
 # Create our conda env
-RUN conda create --name accelerate python=${PYTHON_VERSION} ipython jupyter pip
+RUN conda create --name accelerate python=${PYTHON_VERSION} pip
 # We don't install pytorch here yet since CUDA isn't available
 # instead we use the direct torch wheel
 ENV PATH /opt/conda/envs/accelerate/bin:$PATH