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.
1 parent c35f8b2 commit 9866372Copy full SHA for 9866372
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