ModuleNotFoundError: No module named 'big_vision'

[sycamoreoak]

fresh venv on an up-to-date Ubuntu 22.04.4 LTS

following the README.md:

$ pip install .

$python scenic/main.py -- \
  --config=scenic/projects/baselines/configs/imagenet/imagenet_vit_config.py \
  --workdir=./
2024-06-11 20:31:14.693998: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT
Traceback (most recent call last):
  File "~/src/python/scenic/scenic/main.py", line 27, in <module>
    from scenic.train_lib import trainers
  File "~/src/python/scenic/env_scenic/lib/python3.10/site-packages/scenic/train_lib/trainers.py", line 18, in <module>
    from scenic.train_lib.transfer import transfer_trainer
  File "~/src/python/scenic/env_scenic/lib/python3.10/site-packages/scenic/train_lib/transfer/transfer_trainer.py", line 38, in <module>
    from scenic.train_lib import pretrain_utils
  File "~/src/python/scenic/env_scenic/lib/python3.10/site-packages/scenic/train_lib/pretrain_utils.py", line 23, in <module>
    from big_vision import utils
ModuleNotFoundError: No module named 'big_vision'

i'll go install big_vision but it would be nice if the jax ecosystem had a stronger JustWorks coefficient

[xiexie123]

+1

[ayushnangia]

just install the big_vision github