Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
739c350c
Unverified
Commit
739c350c
authored
Mar 14, 2024
by
陈序
Committed by
GitHub
Mar 13, 2024
Browse files
[Minor Fix] Use cupy-cuda11x in CUDA 11.8 build (#3256)
parent
ba8dc958
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
0 deletions
+6
-0
setup.py
setup.py
+6
-0
No files found.
setup.py
View file @
739c350c
...
...
@@ -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
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment