install cudf 23.12

rapidsai · Jan 24, 2024 · 3ab07c6 · 3ab07c6
1 parent 06afeee
commit 3ab07c6
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/.github/workflows/gpu-ci.yml b/.github/workflows/gpu-ci.yml
@@ -13,7 +13,7 @@ jobs:
   gpu-ci:
     runs-on: linux-amd64-gpu-p100-latest-1
     container:
-      image: nvcr.io/nvidian/crossfit-ci:23.10
+      image: nvcr.io/nvidian/crossfit-ci:23.10-cudf23.12
       env:
         NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
       options: --shm-size=1G
@@ -32,7 +32,7 @@ jobs:
   benchmark:
     runs-on: linux-amd64-gpu-p100-latest-1
     container:
-      image: nvcr.io/nvidian/crossfit-ci:23.10
+      image: nvcr.io/nvidian/crossfit-ci:23.10-cudf23.12
       env:
         NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
       options: --shm-size=1G

diff --git a/docker/ci/Dockerfile b/docker/ci/Dockerfile
@@ -3,6 +3,7 @@ FROM nvcr.io/nvidia/nemo:23.10
 COPY . /tmp/crossfit/
 RUN cd /tmp/crossfit && \
     python3 -m pip install .[pytorch-dev] && \
+    python3 -m pip install -r requirements/gpu.txt --no-cache-dir --extra-index-url https://pypi.nvidia.com && \
     python3 -m pip install beir && \
     rm -r /tmp/crossfit
 

diff --git a/docker/ci/build_and_push.sh b/docker/ci/build_and_push.sh
@@ -6,7 +6,7 @@
 set -e
 
 IMAGE_NAME=nvcr.io/nvidian/crossfit-ci
-IMAGE_TAG=23.10
+IMAGE_TAG=23.10-cudf23.12
 
 docker build -t ${IMAGE_NAME}:${IMAGE_TAG} -f docker/ci/Dockerfile .