Docker AttributeError: module 'jaxlib.pocketfft' has no attribute 'pocketfft
cdmichaelb opened this issue · comments
`Traceback (most recent call last):
File "app.py", line 18, in
from dalle_model import DalleModel
File "/app/dalle_model.py", line 5, in
import jax
File "/usr/local/lib/python3.8/dist-packages/jax/init.py", line 116, in
from .experimental.maps import soft_pmap as soft_pmap
File "/usr/local/lib/python3.8/dist-packages/jax/experimental/maps.py", line 26, in
from .. import numpy as jnp
File "/usr/local/lib/python3.8/dist-packages/jax/numpy/init.py", line 19, in
from . import fft as fft
File "/usr/local/lib/python3.8/dist-packages/jax/numpy/fft.py", line 17, in
from jax._src.numpy.fft import (
File "/usr/local/lib/python3.8/dist-packages/jax/_src/numpy/fft.py", line 19, in
from jax import lax
File "/usr/local/lib/python3.8/dist-packages/jax/lax/init.py", line 332, in
from jax._src.lax.fft import (
File "/usr/local/lib/python3.8/dist-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'`
Same issue here
Still seem to be having this issue while working with MEGA.
Still seem to be having this issue while working with MEGA.
Check #79
I manually changed my Dockerfile to replicate the changes in #79 and this issue still happens on startup.
EDIT: Looks like I forgot to do docker-compose build after changing the Dockerfile.
After I implemented the changes in PR #79, ran docker-compose build, then ran docker-compose up everything started working.