search

borisdayma / dalle-mini

DALL·E Mini - Generate images from a text prompt
https://www.craiyon.com
Apache License 2.0
14.74k stars 1.2k forks source link

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

Open afiqfirdaus opened 2 years ago

afiqfirdaus commented 2 years ago

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 commented 2 years ago

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