search

google / nerfies

This is the code for Deformable Neural Radiance Fields, a.k.a. Nerfies.
https://nerfies.github.io
Apache License 2.0
1.64k stars 221 forks source link

AttributeError: module 'jaxlib.pocketfft' has no attribute 'pocketfft' #64

Open panxiaolong110 opened 2 years ago

panxiaolong110 commented 2 years ago

Traceback (most recent call last): File "D:/VS/python/nerfies-main/train.py", line 23, in from flax import jax_utils File "D:\RJ\A\python3.8\lib\site-packages\flax__init.py", line 36, in from . import core File "D:\RJ\A\python3.8\lib\site-packages\flax\core__init.py", line 15, in from .axes_scan import broadcast File "D:\RJ\A\python3.8\lib\site-packages\flax\core\axes_scan.py", line 17, in import jax File "D:\RJ\A\python3.8\lib\site-packages\jax__init__.py", line 109, in from .experimental.maps import soft_pmap File "D:\RJ\A\python3.8\lib\site-packages\jax\experimental\maps.py", line 25, in from .. import numpy as jnp File "D:\RJ\A\python3.8\lib\site-packages\jax\numpy\init__.py", line 16, in from . import fft File "D:\RJ\A\python3.8\lib\site-packages\jax\numpy\fft.py", line 17, in from jax._src.numpy.fft import ( File "D:\RJ\A\python3.8\lib\site-packages\jax_src\numpy\fft.py", line 19, in from jax import lax File "D:\RJ\A\python3.8\lib\site-packages\jax\lax\init__.py", line 332, in from jax._src.lax.fft import ( File "D:\RJ\A\python3.8\lib\site-packages\jax_src\lax\fft.py", line 145, in xla.backend_specific_translations['cpu'][fft_p] = pocketfft.pocketfft AttributeError: module 'jaxlib.pocketfft' has no attribute 'pocketfft'

CASIATumingfei commented 1 year ago

Hi, I met the same error with you. Did you fix it?

JYC1012 commented 1 year ago

me too