From 93ca2aadd7e2fb12a339e23279a9b6778f7017e2 Mon Sep 17 00:00:00 2001 From: Emilio Castillo Date: Thu, 11 Nov 2021 04:46:39 +0000 Subject: [PATCH] Bump cupy version in Dockerfile to v9.6.0 --- 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 f1bc10a8b92..2f5c247669a 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]==9.5.0" +RUN pip3 install --no-cache-dir "cupy-cuda112[all]==9.6.0" diff --git a/docker/rocm/Dockerfile b/docker/rocm/Dockerfile index 773f2b3d07c..9158849d8fe 100644 --- a/docker/rocm/Dockerfile +++ b/docker/rocm/Dockerfile @@ -13,6 +13,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]==9.5.0" +RUN pip3 install --no-cache-dir "cupy-rocm-4-0[all]==9.6.0" USER rocm-user