diff --git a/ci/base.Dockerfile b/ci/base.Dockerfile index 2de6aa7f64..0b6e9b3f50 100644 --- a/ci/base.Dockerfile +++ b/ci/base.Dockerfile @@ -54,4 +54,4 @@ RUN pyenv update && \ ENV PATH="/root/.pyenv/shims:${PATH}" -RUN pip install --upgrade pip setuptools wheel tox cupy-cuda11x +RUN pip install --upgrade pip setuptools wheel tox cupy-cuda11x==12.3.0