Update HPU Dockerfile to latest version and fix CI (#191)

* Update HPU Dockerfile to latest version and fix CI Signed-off-by: Sun, Xuehao <xuehao.sun@intel.com> * update dockerfile Signed-off-by: Sun, Xuehao <xuehao.sun@intel.com> * fix ci Signed-off-by: Sun, Xuehao <xuehao.sun@intel.com> --------- Signed-off-by: Sun, Xuehao <xuehao.sun@intel.com>
opea-project · Nov 6, 2024 · 6633314 · 6633314
1 parent 9d76832
commit 6633314
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/.github/workflows/docker/hpu.dockerfile b/.github/workflows/docker/hpu.dockerfile
@@ -1,4 +1,4 @@
-FROM vault.habana.ai/gaudi-docker/1.13.0/ubuntu22.04/habanalabs/pytorch-installer-2.1.0:latest as hpu
+FROM vault.habana.ai/gaudi-docker/1.18.0/ubuntu22.04/habanalabs/pytorch-installer-2.4.0:latest as hpu
 
 ENV LANG=en_US.UTF-8
 ENV PYTHONPATH=/root:/usr/lib/habanalabs/

diff --git a/.github/workflows/model_test_cpu.yml b/.github/workflows/model_test_cpu.yml
@@ -13,6 +13,7 @@ on:
       - evals/evaluation/**
       - evals/metrics/**
       - setup.py
+      - "!**.md"
   workflow_dispatch:
 
 # If there is a new commit, the previous jobs will be canceled

diff --git a/.github/workflows/model_test_hpu.yml b/.github/workflows/model_test_hpu.yml
@@ -13,6 +13,7 @@ on:
         - evals/evaluation/**
         - evals/metrics/**
         - setup.py
+        - "!**.md"
   workflow_dispatch:
 
 # If there is a new commit, the previous jobs will be canceled
@@ -60,7 +61,7 @@ jobs:
             docker stop ${{ env.CONTAINER_NAME }}
             docker rm -vf ${{ env.CONTAINER_NAME }} || true
           fi
-          docker run -tid --runtime=habana --name=${{ env.CONTAINER_NAME }} -v ${{ github.workspace }}:/GenAIEval -v /dev/shm:/dev/shm ${{ env.DOCKER_NAME }}:${{ env.DOCKER_TAG }}
+          docker run -tid --runtime=habana -e HABANA_VISIBLE_DEVICES=all --name=${{ env.CONTAINER_NAME }} -v ${{ github.workspace }}:/GenAIEval -v /dev/shm:/dev/shm ${{ env.DOCKER_NAME }}:${{ env.DOCKER_TAG }}
 
       - name: Evaluation
         run: |

diff --git a/docker/hpu.dockerfile b/docker/hpu.dockerfile
@@ -1,4 +1,4 @@
-FROM vault.habana.ai/gaudi-docker/1.16.1/ubuntu22.04/habanalabs/pytorch-installer-2.2.2:latest as hpu
+FROM vault.habana.ai/gaudi-docker/1.18.0/ubuntu22.04/habanalabs/pytorch-installer-2.4.0:latest as hpu
 
 ENV LANG=en_US.UTF-8
 ENV PYTHONPATH=/root:/usr/lib/habanalabs/