Skip to content

Commit 600413c

Browse files
committed
stringify cuda major version in setup.py
1 parent 90de096 commit 600413c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/packaging/libs_wheel/setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323

2424

2525
def get_requirements():
26-
if "##CUDA_MAJOR##" == 13:
26+
if "##CUDA_MAJOR##" == "13":
2727
reqs = ["nvidia-cuda-runtime"]
2828
else:
2929
reqs = ["nvidia-cuda-runtime-cu##CUDA_MAJOR##"]

0 commit comments

Comments
 (0)