requirements-gpu.txt

# General dependencies
agl-report-reader==0.4.0
aiohappyeyeballs==2.4.4
aiohttp==3.11.11
aiosignal==1.3.2
asgiref==3.8.1
attrs==24.3.0
black==24.10.0
cffi==1.17.1
charset-normalizer==3.4.1
cryptography==44.0.0
django==5.1.4
django-bootstrap5==24.3
djangorestframework==3.15.2
faker==33.1.0
ffmpeg-python==0.2.0
filelock==3.16.1
frozenlist==1.5.0
fsspec[http]==2024.12.0
future==1.0.0
gender-guesser==0.4.0
idna==3.10
isort==5.12.0
jinja2==3.1.5
joblib==1.4.2
lightning-utilities==0.11.9
markupsafe==3.0.2
mpmath==1.3.0
multidict==6.1.0
networkx==3.4.2
numpy==2.2.1
opencv-python==4.10.0.84
packaging==24.2
pandas==2.2.3
pdfminer-six==20231228
pdfplumber==0.11.4
pillow==11.0.0
propcache==0.2.1
pycparser==2.22
pypdfium2==4.30.1
pytesseract==0.3.13
python-dateutil==2.9.0.post0
pytorch-lightning==2.5.0.post0
pytz==2024.2
pyyaml==6.0.2
ruff==0.8.6
scikit-learn==1.6.0
scipy==1.14.1
six==1.17.0
sqlparse==0.5.3
sympy==1.13.1
pytesseract==0.3.13
threadpoolctl==3.5.0
torch==2.5.1+cu117  # GPU version of PyTorch with CUDA support
torchmetrics==1.6.1
torchvision==0.20.1+cu117  # GPU version of torchvision
tqdm==4.67.1
triton==3.1.0
typing-extensions==4.12.2
tzdata==2024.2
urllib3==2.3.0
yarl==1.18.3

# CUDA and NVIDIA dependencies
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-nccl-cu12==2.21.5
nvidia-nvjitlink-cu12==12.4.127
nvidia-nvtx-cu12==12.4.127