Use CuPy 13 for NVIDIA NGC Base Image

discovery-unicamp · Sep 18, 2023 · 9fec3e6 · 9fec3e6
1 parent a8a920c
commit 9fec3e6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/build/hpccm/build_docker.py b/build/hpccm/build_docker.py
@@ -70,6 +70,10 @@ def str2bool(string):
 
     if is_devel:
         pip_package_install = ("%s %s" % (pip_package_install, "git+https://github.com/cupy/cupy.git"))
+    else:
+        pip_package_install = ("%s %s" % (pip_package_install, "cupy==13.0.0b1"))
+        Stage0 += shell(commands=["rm -r /usr/local/lib/python3.10/dist-packages/cupy_cuda12x-12.0.0b3.dist-info"]) # not the best solution but it works
+
 
 Stage0 += shell(commands=["pip3 install pip --upgrade"])