No GPU/TPU found, falling back to CPU...

[afiqfirdaus]

while running python3 app.py 8080, i got --> Starting DALL-E Server. This might take up to two minutes. WARNING:absl:No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)

I have pytorch and its dependencies installed with CUDA:

(base) afiq-ubuntu@afiqubuntu-MS-7C96:~$ python3 Python 3.9.12 (main, Apr 5 2022, 06:56:58) [GCC 7.5.0] :: Anaconda, Inc. on linux Type "help", "copyright", "credits" or "license" for more information.

import torch torch.cuda.is_available() True

[SnoopJ]

I experienced this issue as well and was able to solve it by installing a version of jaxlib compatible with my specific CUDA version. See https://github.com/google/jax/issues/5231#issuecomment-748621987