example-submission / requirements.torch.gpu.txt
cc-dsri's picture
Add GPU Dockerfile and requirements
f0d7b8f
raw
history blame
1.58 kB
# This file was autogenerated by uv via the following command:
# uv pip compile --python-version 3.12 -o requirements.torch.gpu.txt requirements.torch.gpu.in
filelock==3.20.0
# via torch
fsspec==2025.10.0
# via torch
jinja2==3.1.6
# via torch
markupsafe==3.0.3
# via jinja2
mpmath==1.3.0
# via sympy
networkx==3.6
# via torch
numpy==2.3.5
# via torchvision
nvidia-cublas-cu12==12.8.4.1
# via
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
# torch
nvidia-cuda-cupti-cu12==12.8.90
# via torch
nvidia-cuda-nvrtc-cu12==12.8.93
# via torch
nvidia-cuda-runtime-cu12==12.8.90
# via torch
nvidia-cudnn-cu12==9.10.2.21
# via torch
nvidia-cufft-cu12==11.3.3.83
# via torch
nvidia-cufile-cu12==1.13.1.3
# via torch
nvidia-curand-cu12==10.3.9.90
# via torch
nvidia-cusolver-cu12==11.7.3.90
# via torch
nvidia-cusparse-cu12==12.5.8.93
# via
# nvidia-cusolver-cu12
# torch
nvidia-cusparselt-cu12==0.7.1
# via torch
nvidia-nccl-cu12==2.27.5
# via torch
nvidia-nvjitlink-cu12==12.8.93
# via
# nvidia-cufft-cu12
# nvidia-cusolver-cu12
# nvidia-cusparse-cu12
# torch
nvidia-nvshmem-cu12==3.3.20
# via torch
nvidia-nvtx-cu12==12.8.90
# via torch
pillow==12.0.0
# via torchvision
setuptools==80.9.0
# via torch
sympy==1.14.0
# via torch
torch==2.9.1
# via
# -r requirements.torch.gpu.in
# torchvision
torchvision==0.24.1
# via -r requirements.torch.gpu.in
triton==3.5.1
# via torch
typing-extensions==4.15.0
# via torch