An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu.
Opened this issue · 2 comments
MNDL-27 commented
MNDL-27 commented
hollowstrawberry commented
The jaxlib thing is fine, you can ignore it