update version of cupy to be 12.0.0 minimum (#899)

NVIDIA-Merlin · Apr 7, 2023 · b5b06dd · b5b06dd
1 parent 46846ed
commit b5b06dd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docker/dockerfile.merlin b/docker/dockerfile.merlin
@@ -104,9 +104,9 @@ RUN pip install --no-cache-dir --upgrade pip; pip install --no-cache-dir "cmake<
 
 # cupy-cuda wheels come from a different URL on aarch64
 RUN if [ $(uname -m) == "aarch64" ] ; then \
-        pip install cupy-cuda115 -f https://pip.cupy.dev/aarch64/ ; \
+        pip install cupy-cuda11x -f https://pip.cupy.dev/aarch64/ ; \
     else \
-        pip install cupy-cuda117 ; \
+        pip install cupy-cuda11x ; \
     fi
 
 # tritonclient[all]==2.29.0: latest tritonclient removes the perf_* binaries, so specified to version 2.29.0