Skip to content

Commit b6655b6

Browse files
authored
Make Python CUDA dependencies resolve for arm64, too (#2501)
See also #1602
1 parent e246521 commit b6655b6

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pyproject.toml

+4-4
Original file line numberDiff line numberDiff line change
@@ -22,10 +22,10 @@ dependencies = [
2222
'numpy >= 1.24',
2323
'scipy >= 1.10.1',
2424
'requests >= 2.31',
25-
'nvidia-cublas-cu12 ~= 12.0; platform_machine == "x86_64"',
26-
'nvidia-cuda-runtime-cu12 ~= 12.0; platform_machine == "x86_64"',
27-
'nvidia-cusolver-cu12 ~= 11.4; platform_machine == "x86_64"',
28-
'nvidia-cuda-nvrtc-cu12 ~= 12.0; platform_machine == "x86_64"'
25+
'nvidia-cublas-cu12 ~= 12.0',
26+
'nvidia-cuda-runtime-cu12 ~= 12.0',
27+
'nvidia-cusolver-cu12 ~= 11.4',
28+
'nvidia-cuda-nvrtc-cu12 ~= 12.0'
2929
]
3030
classifiers = [
3131
'Intended Audience :: Science/Research',

0 commit comments

Comments
 (0)