test

yfukai · Jan 30, 2024 · e608a79 · e608a79
1 parent 64ee888
commit e608a79
Showing 1 changed file with 27 additions and 27 deletions.
diff --git a/.github/workflows/docker-publish.yml b/.github/workflows/docker-publish.yml
@@ -123,31 +123,31 @@ jobs:
 
       # Build and push Docker image with Buildx
       # https://github.com/docker/build-push-action
-      - name: Build and push Docker image (JAX conda CPU)
-        id: build-and-push-jax-conda-cpu
-        uses: docker/build-push-action@v5
-        with:
-          context: jax_image/
-          platforms: linux/amd64,linux/arm64/v8
-          push: true
-          tags: ${{ steps.meta_jax_conda.outputs.tags }}
-          labels: ${{ steps.meta_jax_conda.outputs.labels }}
-          build-args: |
-            JAX_VERSION=${{ env.JAX_VERSION }}
-
-      - name: Build and push Docker image (JAX conda GPU)
-        id: build-and-push-jax-conda-gpu
-        uses: docker/build-push-action@v5
-        with:
-          context: jax_image/
-          platforms: linux/amd64
-          push: true
-          tags: ${{ steps.meta_jax_conda_cuda.outputs.tags }}
-          labels: ${{ steps.meta_jax_conda_cuda.outputs.labels }}
-          build-args: |
-            JAX_VERSION=${{ env.JAX_VERSION }}
-            JAX_VERSION_EXTRA=cuda11_local
-            BASE_IMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
+#      - name: Build and push Docker image (JAX conda CPU)
+#        id: build-and-push-jax-conda-cpu
+#        uses: docker/build-push-action@v5
+#        with:
+#          context: jax_image/
+#          platforms: linux/amd64,linux/arm64/v8
+#          push: true
+#          tags: ${{ steps.meta_jax_conda.outputs.tags }}
+#          labels: ${{ steps.meta_jax_conda.outputs.labels }}
+#          build-args: |
+#            JAX_VERSION=${{ env.JAX_VERSION }}
+#
+#      - name: Build and push Docker image (JAX conda GPU)
+#        id: build-and-push-jax-conda-gpu
+#        uses: docker/build-push-action@v5
+#        with:
+#          context: jax_image/
+#          platforms: linux/amd64
+#          push: true
+#          tags: ${{ steps.meta_jax_conda_cuda.outputs.tags }}
+#          labels: ${{ steps.meta_jax_conda_cuda.outputs.labels }}
+#          build-args: |
+#            JAX_VERSION=${{ env.JAX_VERSION }}
+#            JAX_VERSION_EXTRA=cuda11_local
+#            BASE_IMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
 
       - name: Build and push Docker image (CPU)
         id: build-and-push-cpu
@@ -159,7 +159,7 @@ jobs:
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
           build-args: |
-            BASE_IMAGE=${{ env.REGISTRY }}/${{ env.JAX_CONDA_IMAGE_NAME }}:${{ env.meta_jax_conda_first_tag }}
+            BASE_IMAGE=${{ env.meta_jax_conda_first_tag }}
 
       - name: Build and push Docker image (GPU)
         id: build-and-push-gpu
@@ -171,4 +171,4 @@ jobs:
           tags: ${{ steps.meta_cuda.outputs.tags }}
           labels: ${{ steps.meta_cuda.outputs.labels }}
           build-args: |
-            BASE_IMAGE=${{ env.REGISTRY }}/${{ env.JAX_CONDA_IMAGE_NAME }}:${{ env.meta_jax_conda_cuda_first_tag }}
+            BASE_IMAGE=${{ env.meta_jax_conda_cuda_first_tag }}