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

[panxiaolong110]

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]

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

[JYC1012]

me too