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