requirements.txt

antlr4-python3-runtime==4.9.3
blessed==1.20.0
contourpy==1.3.1
cycler==0.12.1
filelock==3.16.1
fonttools==4.55.3
fsspec==2024.10.0
#gpustat==1.1.1
hydra-core==1.3.2
iopath==0.1.10
Jinja2==3.1.4
kiwisolver==1.4.7
MarkupSafe==3.0.2
matplotlib==3.9.4
mpmath==1.3.0
networkx==3.4.2
numpy==1.26.4
#nvidia-cublas-cu12==12.4.5.8
#nvidia-cuda-cupti-cu12==12.4.127
#nvidia-cuda-nvrtc-cu12==12.4.127
#nvidia-cuda-runtime-cu12==12.4.127
#nvidia-cudnn-cu12==9.1.0.70
#nvidia-cufft-cu12==11.2.1.3
#nvidia-curand-cu12==10.3.5.147
#nvidia-cusolver-cu12==11.6.1.9
#nvidia-cusparse-cu12==12.3.1.170
#nvidia-ml-py==12.560.30
#nvidia-nccl-cu12==2.21.5
#nvidia-nvjitlink-cu12==12.4.127
#nvidia-nvtx-cu12==12.4.127
omegaconf==2.3.0
opencv-python==4.10.0.84
packaging==24.2
pillow==11.0.0
portalocker==3.0.0
psutil==6.1.0
pycocotools==2.0.8
pyparsing==3.2.0
python-dateutil==2.9.0.post0
PyYAML==6.0.2
sam2==1.1.0
six==1.17.0
sympy==1.13.1
#torch==2.5.1
#torchaudio==2.5.1
#torchvision==0.20.1
tqdm==4.67.1
#triton==3.1.0
typing_extensions==4.12.2
wcwidth==0.2.13