Open neilbGH opened 2 years ago
I reinstalled everything, including video card drivers and docker and it's working now.
@neilbGH are you able to step through the code with your setup?
I run this project on linux , how to run it on GPU? I had the follow message.
WARNING:jax._src.lib.xla_bridge:No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)
Fortunately, I have resoled it with method below, :
pip install --upgrade pip
pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
jax 、jaxlib、cuda what is the corresponding version
Getting the following trying to run dalle-mini in Docker on Windows with an RTX3090 and 32GB of RAM. nvidia-smi shows the GPU is detected and I've run other GPU Docker images without issue.