Jax Installation about d2nerf HOT 3 CLOSED

Hi authors, thanks for releasing the code! I am currently facing issues while running this repo on GPU (it's working on CPU but it takes forever). I think the problem is with the required version of jax and cuda support.

Can you please provide more details regarding the CUDA version and NVIDIA drivers that you are using for jax 0.2.26? TIA

hi, i had solved the problem.here is my solution that may help you:
To keep the version with the same as hypernerf:
pip install --upgrade jax==0.2.20 jaxlib==0.1.71+cuda111 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
pip install flax==0.3.6

And be careful this command which can upgrade the jax and jaxlib:
pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_releases.html

By the way, you can execute the command to verify：
python -c "import jax;print(jax.devices())"
if work, it will print like this:
[GpuDevice(id=0, process_index=0), GpuDevice(id=1, process_index=0), GpuDevice(id=2, process_index=0), GpuDevice(id=3, process_index=0)]

from d2nerf.

I am having the same problem.

from d2nerf.

Hi, thank you for your interest in our work! Our environment should be exactly the same as HyperNeRF. For CUDA we used 11.3, and for Jax we used 0.2.27, but we are aware that this specific Jax version seems to have been removed from the Jax wheel. Did you encounter any issues when installing the environment?

from d2nerf.