diff --git a/docker/python3/Dockerfile b/docker/python3/Dockerfile index d4cd1cdf455..725d02f4565 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 "cupy-cuda112[all]==10.0.0a2" +RUN pip3 install --no-cache-dir "cupy-cuda112[all]==10.0.0b1" diff --git a/docker/rocm/Dockerfile b/docker/rocm/Dockerfile index a2932bd04e4..8ca0aa07b52 100644 --- a/docker/rocm/Dockerfile +++ b/docker/rocm/Dockerfile @@ -12,6 +12,6 @@ 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 "cupy-rocm-4-0[all]==10.0.0a2" +RUN pip3 install --no-cache-dir "cupy-rocm-4-0[all]==10.0.0b1" USER rocm-user