From a977e9f0a0e4184d23f7ee01ac5c926d0e8027b7 Mon Sep 17 00:00:00 2001 From: Emilio Castillo Date: Thu, 11 Nov 2021 04:44:55 +0000 Subject: [PATCH] Bump cupy version in Dockerfile to v10.0.0rc1 --- docker/python3/Dockerfile | 2 +- docker/rocm/Dockerfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/python3/Dockerfile b/docker/python3/Dockerfile index 8e471c8ccd6..65b47b90b6b 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://github.com/cupy/cupy/releases/v10.0.0b3 "cupy-cuda114[all]==10.0.0b3" +RUN pip3 install --no-cache-dir -f https://github.com/cupy/cupy/releases/v10.0.0rc1 "cupy-cuda114[all]==10.0.0rc1" diff --git a/docker/rocm/Dockerfile b/docker/rocm/Dockerfile index 00decb9adec..3c695c6831a 100644 --- a/docker/rocm/Dockerfile +++ b/docker/rocm/Dockerfile @@ -16,6 +16,6 @@ RUN apt-get update -y && \ RUN curl https://bootstrap.pypa.io/get-pip.py | sudo python3.7 RUN pip3 install --no-cache-dir -U install setuptools pip -RUN pip3 install --no-cache-dir -f https://github.com/cupy/cupy/releases/v10.0.0b3 "cupy-rocm-4-3[all]==10.0.0b3" +RUN pip3 install --no-cache-dir -f https://github.com/cupy/cupy/releases/v10.0.0rc1 "cupy-rocm-4-3[all]==10.0.0rc1" USER rocm-user