multinerf icon indicating copy to clipboard operation
multinerf copied to clipboard

Published 20 hours ago • verified publisher icon google-research

No GPU/TPU found

Open basit-7 opened this issue 1 year ago • 5 comments

When I run all unit tests, I get the following error. image

I am working with Quadro RTX 8000, cuda 11.0.

basit-7 avatar Sep 15 '22 06:09 basit-7

I'd check out the Jax installation docs to see what their guidance is. You'll probably need to reinstall Jax with CUDA/GPU support enabled.

jonbarron avatar Sep 15 '22 15:09 jonbarron

@basit-7

https://github.com/google/jax#installation

just install jax with gpu support:

pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

HannahHaensen avatar Sep 21 '22 11:09 HannahHaensen

After installation, mine failed on test script

Derry-Xing avatar Oct 03 '22 12:10 Derry-Xing