search

google-research / multinerf

A Code Release for Mip-NeRF 360, Ref-NeRF, and RawNeRF
Apache License 2.0
3.58k stars 339 forks source link

No GPU/TPU found #46

Open basit-7 opened 1 year ago

basit-7 commented 1 year ago

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

image

I am working with Quadro RTX 8000, cuda 11.0.

jonbarron commented 1 year ago

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.

HannahHaensen commented 1 year ago

@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

Derry-Xing commented 1 year ago

After installation, mine failed on test script

Jamie725 commented 1 year ago

Same here... does anyone have any solutions for it? Thanks!

lozhn commented 1 year ago

pip install --upgrade "jax[cuda11_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html this worked for me