diff --git a/docker/python3/Dockerfile b/docker/python3/Dockerfile index 39bb91f74be..14994d5b107 100644 --- a/docker/python3/Dockerfile +++ b/docker/python3/Dockerfile @@ -9,4 +9,4 @@ RUN apt-get update -y && \ rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/* RUN pip3 install --no-cache-dir -U install setuptools pip -RUN pip3 install --no-cache-dir -f https://pip.cupy.dev/pre/ "cupy-cuda11x[all]==12.0.0b3" +RUN pip3 install --no-cache-dir -f https://pip.cupy.dev/pre/ "cupy-cuda11x[all]==12.0.0rc1" diff --git a/docker/rocm/Dockerfile b/docker/rocm/Dockerfile index 184b4c7f165..9789327475a 100644 --- a/docker/rocm/Dockerfile +++ b/docker/rocm/Dockerfile @@ -14,6 +14,6 @@ RUN apt-get update -y && \ update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.6 1 RUN python3 -m pip install --no-cache-dir -U install setuptools pip -RUN python3 -m pip install --no-cache-dir -f https://pip.cupy.dev/pre/ "cupy-rocm-5-0[all]==12.0.0b3" +RUN python3 -m pip install --no-cache-dir -f https://pip.cupy.dev/pre/ "cupy-rocm-5-0[all]==12.0.0rc1" USER rocm-user