pytorch_template/requirements.txt at main · Axect/pytorch_template · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
alembic==1.18.4
annotated-doc==0.0.4
annotated-types==0.7.0
beaupy==3.11.0
certifi==2026.2.25
charset-normalizer==3.4.5
click==8.3.1
colorlog==6.10.1
contourpy==1.3.3
cuda-bindings==12.9.4
cuda-pathfinder==1.4.1
cycler==0.12.1
emoji==2.15.0
filelock==3.25.0
fireducks==1.4.4
firefw==1.4.4
fonttools==4.61.1
fsspec==2026.2.0
gitdb==4.0.12
gitpython==3.1.46
greenlet==3.3.2
idna==3.11
iniconfig==2.3.0
jinja2==3.1.6
kiwisolver==1.5.0
mako==1.3.10
markdown-it-py==4.0.0
markupsafe==3.0.3
matplotlib==3.10.8
mdurl==0.1.2
mpmath==1.3.0
networkx==3.6.1
numpy==2.4.3
nvidia-cublas-cu12==12.8.4.1
nvidia-cuda-cupti-cu12==12.8.90
nvidia-cuda-nvrtc-cu12==12.8.93
nvidia-cuda-runtime-cu12==12.8.90
nvidia-cudnn-cu12==9.10.2.21
nvidia-cufft-cu12==11.3.3.83
nvidia-cufile-cu12==1.13.1.3
nvidia-curand-cu12==10.3.9.90
nvidia-cusolver-cu12==11.7.3.90
nvidia-cusparse-cu12==12.5.8.93
nvidia-cusparselt-cu12==0.7.1
nvidia-nccl-cu12==2.27.5
nvidia-nvjitlink-cu12==12.8.93
nvidia-nvshmem-cu12==3.4.5
nvidia-nvtx-cu12==12.8.90
optuna==4.7.0
packaging==26.0
pandas==2.3.3
pillow==12.1.1
platformdirs==4.9.4
pluggy==1.6.0
protobuf==6.33.5
pyarrow==21.0.0
pydantic==2.12.5
pydantic-core==2.41.5
pygments==2.19.2
pyparsing==3.3.2
pytest==9.0.2
python-dateutil==2.9.0.post0
python-yakh==0.4.1
pytorch-optimizer==3.10.0
pytorch-scheduler==0.2.1
pytz==2026.1.post1
pyyaml==6.0.3
questo==0.4.2
requests==2.32.5
rich==14.3.3
scienceplots==2.2.1
sentry-sdk==2.54.0
setuptools==82.0.1
shellingham==1.5.4
six==1.17.0
smmap==5.0.3
sqlalchemy==2.0.48
sympy==1.14.0
torch==2.10.0
tqdm==4.67.3
triton==3.6.0
typer==0.24.1
typing-extensions==4.15.0
typing-inspection==0.4.2
tzdata==2025.3
urllib3==2.6.3
wandb==0.25.0