Giters

google-research / scenic

Scenic: A Jax Library for Computer Vision Research and Beyond

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Requirements for Vid2Seq

lawrenceztang opened this issue · comments

commented

I'm having a lot of trouble running Vid2Seq. Here are my steps:

  1. Create new TPU v2-8
  2. gcloud compute tpus tpu-vm ssh node-1 --zone=us-central1-f
  3. Python version is 3.8, so update Python 3.10. Install dependencies:
sudo apt update; sudo apt install build-essential libssl-dev zlib1g-dev \
libbz2-dev libreadline-dev libsqlite3-dev curl \
libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev
  1. Install Python 3.10:
curl https://pyenv.run | bash
echo 'export PATH="$HOME/.pyenv/bin:$PATH"' >> ~/.bashrc
echo 'eval "$(pyenv init -)"' >> ~/.bashrc
echo 'eval "$(pyenv virtualenv-init -)"' >> ~/.bashrc
source ~/.bashrc
pyenv install 3.10.7
pyenv global 3.10.7
  1. Pip can't find t5x so install t5x:
git clone --branch=main https://github.com/google-research/t5x
cd t5x
python3 -m pip install -e '.[tpu]' -f \
https://storage.googleapis.com/jax-releases/libtpu_releases.html
  1. git clone https://github.com/google-research/scenic
  2. pip install -r scenic/projects/vid2seq/requirements.txt
  3. I get ERROR: Cannot install flax==0.5.0 and t5x==0.0.0 because these package versions have conflicting dependencies.

Now what do I do?