Closed Schortenger closed 1 year ago
FYI, "FAILED_PRECONDITION: DNN library initialization failed" will appear if we use cuda11 with Jax==0.4.8. The solution would be using conda install -c "nvidia/label/cuda-12.1.0" cuda-nvcc to install cuda12, and then pip install jaxlib==0.4.7.
conda install -c "nvidia/label/cuda-12.1.0" cuda-nvcc
pip install jaxlib==0.4.7
FYI, "FAILED_PRECONDITION: DNN library initialization failed" will appear if we use cuda11 with Jax==0.4.8. The solution would be using
conda install -c "nvidia/label/cuda-12.1.0" cuda-nvcc
to install cuda12, and thenpip install jaxlib==0.4.7
.