jaxlib.xla_extension.XlaRuntimeError: UNIMPLEMENTED: DNN library is not found.

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

jaxlib.xla_extension.XlaRuntimeError: UNIMPLEMENTED: DNN library is not found. #52

Closed yao990113 closed 1 year ago

yao990113 commented 1 year ago

I have jax 0.3.17 and jaxlib 0.3.15 for cudnn82+cuda11. , my version of cudnn is 8.2.0 , but I still have problem as title ,can you help me? thank you

za970120604 commented 1 year ago

hello , @yao990113 , may I ask how did you solve this problem?