torch
torchvision
torchmetrics
fvcore
iopath
xformers
calflops
submitit
addict
numpy
packaging
Pillow
pyyaml
regex
yapf
arrow
codecarbon
ffmpeg-python
h5py
kaleido
matplotlib
opencv-python
openmim
pandas
plotly
psutil
regex
scikit-learn
scipy
seaborn
tqdm
wandb
setuptools==80.2.0
nbformat>=4.2.0
--extra-index-url https://download.pytorch.org/whl/cu128
--extra-index-url https://pypi.nvidia.com
