train_dreambooth_flax not found gpu error

[rahilparsana]

No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.) , i have the server with gpu its A100 80gb

[pcuenca]

Hi @rahilparsana, please make sure you follow the JAX installation instructions for CUDA: https://github.com/google/jax#pip-installation-gpu-cuda

[github-actions[bot]]

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.