modrec-workflow/requirements.txt
2025-06-17 15:47:36 -04:00

45 lines
926 B
Plaintext

aiohappyeyeballs~=2.6.1
aiohttp~=3.9.0
aiosignal~=1.3.2
attrs~=25.3.0
filelock~=3.18.0
frozenlist~=1.7.0
fsspec~=2025.5.1
h5py~=3.14.0
idna~=3.10
Jinja2~=3.1.6
lightning~=2.5.1.post0
lightning-utilities~=0.14.3
MarkupSafe~=3.0.2
mpmath~=1.3.0
multidict~=6.4.4
networkx~=3.5
numpy~=2.3.0
nvidia-cublas-cu12~=12.6.4.1
nvidia-cuda-cupti-cu12~=12.6.80
nvidia-cuda-nvrtc-cu12~=12.6.77
nvidia-cuda-runtime-cu12~=12.6.77
nvidia-cudnn-cu12~=9.5.1.17
nvidia-cufft-cu12~=11.3.0.4
nvidia-cufile-cu12~=1.11.1.6
nvidia-curand-cu12~=10.3.7.77
nvidia-cusolver-cu12~=11.7.1.2
nvidia-cusparse-cu12~=12.5.4.2
nvidia-cusparselt-cu12~=0.6.3
nvidia-nccl-cu12~=2.26.2
nvidia-nvjitlink-cu12~=12.6.85
nvidia-nvtx-cu12~=12.6.77
packaging~=24.2
propcache~=0.3.2
pytorch-lightning~=2.5.1.post0
PyYAML~=6.0.2
setuptools~=80.9.0
sympy~=1.14.0
torch~=2.7.1
torchmetrics~=1.7.3
tqdm~=4.67.1
triton~=3.3.1
typing_extensions~=4.14.0
utils~=1.0.2
yarl~=1.20.1