decord
einops
lm_eval
numpy
packaging
pytest
setuptools
timm
transformers
wheel
--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.1.1
torchvision==0.16.1