bump cupy version in Dockerfile to v9.5.0

pnunna93 · Sep 30, 2021 · 281e5e9 · 281e5e9
1 parent 74e8baa
commit 281e5e9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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.4.0"
+RUN pip3 install --no-cache-dir "cupy-cuda112[all]==9.5.0"
diff --git 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.4.0"
+RUN pip3 install --no-cache-dir "cupy-rocm-4-0[all]==9.5.0"
 
 USER rocm-user