We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 385e2f4 + 9866372 commit 6d4278dCopy full SHA for 6d4278d
jax_plugins/cuda/plugin_setup.py
@@ -55,7 +55,7 @@ def has_ext_modules(self):
55
'with_cuda': [
56
"nvidia-cublas-cu12>=12.1.3.1",
57
"nvidia-cuda-cupti-cu12>=12.1.105",
58
- "nvidia-cuda-nvcc-cu12>=12.1.105",
+ "nvidia-cuda-nvcc-cu12>=12.6.85",
59
"nvidia-cuda-runtime-cu12>=12.1.105",
60
"nvidia-cudnn-cu12>=9.1,<10.0",
61
"nvidia-cufft-cu12>=11.0.2.54",
0 commit comments