search

ChikaYan / d2nerf

Apache License 2.0
181 stars 14 forks source link

Jax Installation #10

Closed Achleshwar closed 5 months ago

Achleshwar commented 1 year ago

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

m-richa commented 1 year ago

I am having the same problem.

ChikaYan commented 1 year ago

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?

xjli360 commented 10 months ago

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)]