【fixed】No GPU/TPU found, falling back to CPU.或者XlaRuntimeErrorFAILED_PRECONDITION: DNN library initi

「已注销」

已于 2023-05-23 11:11:55 修改

阅读量3.4k

点赞数 8

文章标签： gpu linux python

于 2023-05-15 12:57:17 首次发布

本文链接：https://blog.csdn.net/m0_48945359/article/details/130682186

版权

问题

用jax报错No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)

或者
XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed.

解决

pip install --upgrade pip

# CUDA 12 installation
# Note: wheels only available on linux.
pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

# CUDA 11 installation
# Note: wheels only available on linux.
pip install --upgrade "jax[cuda11_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html