More tweaks

huggingface · Aug 31, 2023 · 71eb389 · 71eb389
1 parent 79c11e2
commit 71eb389
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 26 deletions.
diff --git a/.github/workflows/build_docker_images.yml b/.github/workflows/build_docker_images.yml
@@ -11,32 +11,32 @@ concurrency:
   cancel-in-progress: false
 
 jobs:
-  latest-cpu:
-    name: "Latest Accelerate CPU [dev]"
-    runs-on: ubuntu-latest
-    steps:
-      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v2
-      - name: Check out code
-        uses: actions/checkout@v3
-      - name: Login to DockerHub
-        uses: docker/login-action@v2
-        with:
-          username: ${{ secrets.DOCKERHUB_USERNAME }}
-          password: ${{ secrets.DOCKERHUB_PASSWORD }}
+  # latest-cpu:
+  #   name: "Latest Accelerate CPU [dev]"
+  #   runs-on: ubuntu-latest
+  #   steps:
+  #     - name: Set up Docker Buildx
+  #       uses: docker/setup-buildx-action@v2
+  #     - name: Check out code
+  #       uses: actions/checkout@v3
+  #     - name: Login to DockerHub
+  #       uses: docker/login-action@v2
+  #       with:
+  #         username: ${{ secrets.DOCKERHUB_USERNAME }}
+  #         password: ${{ secrets.DOCKERHUB_PASSWORD }}
 
-      - name: Free disk space
-        run: |
-          sudo swapoff -a
-          sudo rm -f /swapfile
-          sudo apt clean
+  #     - name: Free disk space
+  #       run: |
+  #         sudo swapoff -a
+  #         sudo rm -f /swapfile
+  #         sudo apt clean
 
-      - name: Build and Push CPU
-        uses: docker/build-push-action@v4
-        with:
-          context: ./docker/accelerate-cpu
-          push: true
-          tags: huggingface/accelerate-cpu
+  #     - name: Build and Push CPU
+  #       uses: docker/build-push-action@v4
+  #       with:
+  #         context: ./docker/accelerate-cpu
+  #         push: true
+  #         tags: huggingface/accelerate-cpu
 
   latest-cuda:
     name: "Latest Accelerate GPU [dev]"

diff --git a/docker/accelerate-gpu/Dockerfile b/docker/accelerate-gpu/Dockerfile
@@ -16,8 +16,8 @@ RUN wget \
     && rm -r Miniconda3-latest-Linux-x86_64.sh
 
 # Create our conda env
-RUN conda create --name accelerate python=${PYTHON_VERSION} ipython jupyter pip
-ENV PATH /opt/conda/envs/accelerate/bin:$PATH
+RUN source activate && \
+    conda create --name accelerate python=${PYTHON_VERSION} ipython jupyter pip
 # Activate the conda env and install torch + accelerate
 RUN source activate accelerate && \
     python3 -m pip install --no-cache-dir \