ModuleNotFoundError: No module named 'compressai._CXX'

[naga-k]

Hi Jason, Thanks for your work.

I am trying to run the chair script and I see this error

(/scratch/jacobcha/nk643/NeRFCodec_public/env) (env) [nk643@n0002 NeRFCodec_public]$ bash scripts/tensorf_train/run_chair.sh
Traceback (most recent call last):
  File "/mmfs1/scratch/jacobcha/nk643/NeRFCodec_public/train.py", line 12, in <module>
    from renderer import *
  File "/mmfs1/scratch/jacobcha/nk643/NeRFCodec_public/renderer.py", line 5, in <module>
    from dataLoader.ray_utils import get_rays
  File "/mmfs1/scratch/jacobcha/nk643/NeRFCodec_public/dataLoader/__init__.py", line 23, in <module>
    from .scene_dataset import SceneDataset
  File "/mmfs1/scratch/jacobcha/nk643/NeRFCodec_public/dataLoader/scene_dataset.py", line 21, in <module>
    from models.tensoRF import TensorVM, TensorCP, raw2alpha, TensorVMSplit, AlphaGridMask
  File "/mmfs1/scratch/jacobcha/nk643/NeRFCodec_public/models/tensoRF.py", line 8, in <module>
    from compressai.ops import compute_padding
  File "/mmfs1/scratch/jacobcha/nk643/NeRFCodec_public/compressai/__init__.py", line 33, in <module>
    from compressai import (
  File "/mmfs1/scratch/jacobcha/nk643/NeRFCodec_public/compressai/entropy_models/__init__.py", line 30, in <module>
    from .entropy_models import EntropyBottleneck, EntropyModel, GaussianConditional
  File "/mmfs1/scratch/jacobcha/nk643/NeRFCodec_public/compressai/entropy_models/entropy_models.py", line 42, in <module>
    from compressai._CXX import pmf_to_quantized_cdf as _pmf_to_quantized_cdf
ModuleNotFoundError: No module named 'compressai._CXX'

As I undersand based on the project structure, I do not need to install compressai directly?

Thank you!

[naga-k]

It worked after I installed CompressAI from the source. I had to clone it in a different folder because of a name conflict with the compressai folder and then install it from the source.

Maybe you need to add compressai to the install instructions and have the users build it from source since the precompiled binaries from pip install might not get recognized.

Thank you!