You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
We solved this issue by upgrading CUDA on our machine.
NVIDIA-SMI 550.54.14 Driver Version: 550.54.14 CUDA Version: 12.4
If someone else has the same issue and can't upgrade, I recommend specifying cuDNN and CUDA Toolkit versions before installing, such as by running the following.
Description
Issue installing JAX on CentOS 7. I have tried both the cuda11 and cuda12 pip installations of JAX. Looks similar to but distinct from #17730.
Error when using JAX after either installation:
Any help would be much appreciated.
System info (python version, jaxlib version, accelerator, etc.)
Nvidia-smi:
NVIDIA-SMI 530.30.02 Driver Version: 530.30.02 CUDA Version: 12.1
pip list
after installing cuda12:pip list
after installing cuda11:The text was updated successfully, but these errors were encountered: