From 739c350c1926682f435316294491aa54661849b6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=99=88=E5=BA=8F?= Date: Thu, 14 Mar 2024 00:43:24 +0800 Subject: [PATCH] [Minor Fix] Use cupy-cuda11x in CUDA 11.8 build (#3256) --- setup.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/setup.py b/setup.py index 023c3cde1910c..accf6bb400310 100644 --- a/setup.py +++ b/setup.py @@ -431,6 +431,12 @@ def get_requirements() -> List[str]: else: with open(get_path("requirements.txt")) as f: requirements = f.read().strip().split("\n") + if nvcc_cuda_version <= Version("11.8"): + # replace cupy-cuda12x with cupy-cuda11x for cuda 11.x + for i in range(len(requirements)): + if requirements[i].startswith("cupy-cuda12x"): + requirements[i] = "cupy-cuda11x" + break return requirements