diff --git a/raft-ann-bench/cpu/Dockerfile b/raft-ann-bench/cpu/Dockerfile index 5f383aaf..59aca7c6 100644 --- a/raft-ann-bench/cpu/Dockerfile +++ b/raft-ann-bench/cpu/Dockerfile @@ -1,7 +1,7 @@ # syntax=docker/dockerfile:1 ARG PYTHON_VER=unset -ARG RAPIDS_VER=unset +ARG RAPIDS_VER=24.10 FROM condaforge/mambaforge:23.3.1-0 AS raft-ann-bench-cpu ARG RAPIDS_VER diff --git a/raft-ann-bench/gpu/Dockerfile b/raft-ann-bench/gpu/Dockerfile index 54de8c73..6022f6b8 100644 --- a/raft-ann-bench/gpu/Dockerfile +++ b/raft-ann-bench/gpu/Dockerfile @@ -4,7 +4,7 @@ ARG CUDA_VER=unset ARG PYTHON_VER=unset ARG LINUX_VER=unset -ARG RAPIDS_VER=unset +ARG RAPIDS_VER=24.10 FROM rapidsai/miniforge-cuda:cuda${CUDA_VER}-base-${LINUX_VER}-py${PYTHON_VER} AS raft-ann-bench ARG CUDA_VER