mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #16138 from hawkinsp:cudnn

Browse Source 
PiperOrigin-RevId: 535367462
This commit is contained in:
jax authors 2023-05-25 13:40:34 -07:00
commit 14089fb2f8
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jaxlib/setup.py
View File

@ -53,7 +53,7 @@ setup(
"nvidia-cuda-cupti-cu11>=11.8",
"nvidia-cuda-nvcc-cu11>=11.8",
"nvidia-cuda-runtime-cu11>=11.8",
"nvidia-cudnn-cu11>=8.6",
"nvidia-cudnn-cu11>=8.8",
"nvidia-cufft-cu11>=10.9",
"nvidia-cusolver-cu11>=11.4",
"nvidia-cusparse-cu11>=11.7",
@ -63,7 +63,7 @@ setup(
"nvidia-cuda-cupti-cu12",
"nvidia-cuda-nvcc-cu12",
"nvidia-cuda-runtime-cu12",
"nvidia-cudnn-cu12",
"nvidia-cudnn-cu12>=8.8",
"nvidia-cufft-cu12",
"nvidia-cusolver-cu12",
"nvidia-cusparse-cu12",

4
setup.py
View File

@ -100,7 +100,7 @@ setup(
"nvidia-cuda-cupti-cu11>=11.8",
"nvidia-cuda-nvcc-cu11>=11.8",
"nvidia-cuda-runtime-cu11>=11.8",
"nvidia-cudnn-cu11>=8.6",
"nvidia-cudnn-cu11>=8.8",
"nvidia-cufft-cu11>=10.9",
"nvidia-cusolver-cu11>=11.4",
"nvidia-cusparse-cu11>=11.7",
@ -112,7 +112,7 @@ setup(
"nvidia-cuda-cupti-cu12",
"nvidia-cuda-nvcc-cu12",
"nvidia-cuda-runtime-cu12",
"nvidia-cudnn-cu12",
"nvidia-cudnn-cu12>=8.8",
"nvidia-cufft-cu12",
"nvidia-cusolver-cu12",
"nvidia-cusparse-cu12",