mirror of https://github.com/vllm-project/vllm
[Minor Fix] Use cupy-cuda11x in CUDA 11.8 build (#3256)
This commit is contained in:
parent
ba8dc958a3
commit
739c350c19
6
setup.py
6
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
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue