From 9fec3e637957da060a91bd969108480d12abf423 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Ser=C3=B3dio?= Date: Mon, 18 Sep 2023 12:04:04 -0300 Subject: [PATCH] Use CuPy 13 for NVIDIA NGC Base Image --- build/hpccm/build_docker.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/build/hpccm/build_docker.py b/build/hpccm/build_docker.py index 6c2c4df..b29cb8e 100644 --- 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"])