Closed shangyuna closed 1 year ago
Python version:3.9.12
CUDA version:11.3.109
CUDNN version:8.2.4
jax version: 0.3.15
jaxlib version: 0.3.15+cuda11.cudann82
GPU: RTX 3070
I ran into the same issue and solved it by installing CUDNN.
I ran into the same issue and solved it by installing CUDNN.
I have installed CUDNN ,version:8.2.4
It works with cudnn 8.2 only