sxyu / plenoctree

PlenOctrees: NeRF-SH Training & Conversion
https://alexyu.net/plenoctrees
BSD 2-Clause "Simplified" License
425 stars 78 forks source link

Invoked in extraction #28

Open zhywanna opened 2 years ago

zhywanna commented 2 years ago

I run the script in MISC to Project Vanilla NeRF to PlenOctree an error happened

image

and I finally find out only running this line

https://github.com/sxyu/plenoctree/blob/f0b82631199a1aa7dc9ce263c08980a3a7504014/octree/nerf/models.py#L70

import this package will get that error.

Anyone knows what happened?

Learningm commented 2 years ago

Probably there exists some conflict between your cuda version and jaxlib. I have met similar error as yours.

For stable jaxlib, I suggest you to use cuda11.1+.