forked from mlech26l/ncps
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrequirements_gpu.txt
63 lines (62 loc) · 10.8 KB
/
requirements_gpu.txt
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
--extra-index-url https://download.pytorch.org/whl/cu124
aiohappyeyeballs==2.4.0 ; python_version >= "3.10" and python_version < "3.11"
aiohttp==3.10.5 ; python_version >= "3.10" and python_version < "3.11"
aiosignal==1.3.1 ; python_version >= "3.10" and python_version < "3.11"
async-timeout==4.0.3 ; python_version >= "3.10" and python_version < "3.11"
attrs==24.2.0 ; python_version >= "3.10" and python_version < "3.11"
colorama==0.4.6 ; python_version >= "3.10" and python_version < "3.11" and platform_system == "Windows"
contourpy==1.3.0 ; python_version >= "3.10" and python_version < "3.11"
cramjam==2.8.3 ; python_version >= "3.10" and python_version < "3.11"
cycler==0.12.1 ; python_version >= "3.10" and python_version < "3.11"
fastparquet==2024.5.0 ; python_version >= "3.10" and python_version < "3.11"
filelock==3.15.4 ; python_version >= "3.10" and python_version < "3.11"
fonttools==4.53.1 ; python_version >= "3.10" and python_version < "3.11"
frozenlist==1.4.1 ; python_version >= "3.10" and python_version < "3.11"
fsspec==2024.9.0 ; python_version >= "3.10" and python_version < "3.11"
fsspec[http]==2024.9.0 ; python_version >= "3.10" and python_version < "3.11"
future==1.0.0 ; python_version >= "3.10" and python_version < "3.11"
idna==3.8 ; python_version >= "3.10" and python_version < "3.11"
jinja2==3.1.4 ; python_version >= "3.10" and python_version < "3.11"
joblib==1.4.2 ; python_version >= "3.10" and python_version < "3.11"
kiwisolver==1.4.7 ; python_version >= "3.10" and python_version < "3.11"
lightning-utilities==0.11.7 ; python_version >= "3.10" and python_version < "3.11"
markupsafe==2.1.5 ; python_version >= "3.10" and python_version < "3.11"
matplotlib==3.9.2 ; python_version >= "3.10" and python_version < "3.11"
mpmath==1.3.0 ; python_version >= "3.10" and python_version < "3.11"
multidict==6.0.5 ; python_version >= "3.10" and python_version < "3.11"
networkx==3.3 ; python_version >= "3.10" and python_version < "3.11"
numpy==1.26.4 ; python_version >= "3.10" and python_version < "3.11"
nvidia-cublas-cu12==12.4.2.65 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "3.11"
nvidia-cuda-cupti-cu12==12.4.99 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "3.11"
nvidia-cuda-nvrtc-cu12==12.4.99 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "3.11"
nvidia-cuda-runtime-cu12==12.4.99 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "3.11"
nvidia-cudnn-cu12==9.1.0.70 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "3.11"
nvidia-cufft-cu12==11.2.0.44 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "3.11"
nvidia-curand-cu12==10.3.5.119 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "3.11"
nvidia-cusolver-cu12==11.6.0.99 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "3.11"
nvidia-cusparse-cu12==12.3.0.142 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "3.11"
nvidia-nccl-cu12==2.20.5 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "3.11"
nvidia-nvjitlink-cu12==12.4.99 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "3.11"
nvidia-nvtx-cu12==12.4.99 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "3.11"
packaging==24.1 ; python_version >= "3.10" and python_version < "3.11"
pandas==2.2.2 ; python_version >= "3.10" and python_version < "3.11"
pillow==10.4.0 ; python_version >= "3.10" and python_version < "3.11"
pyparsing==3.1.4 ; python_version >= "3.10" and python_version < "3.11"
python-dateutil==2.9.0.post0 ; python_version >= "3.10" and python_version < "3.11"
pytorch-lightning==2.4.0 ; python_version >= "3.10" and python_version < "3.11"
pytz==2024.1 ; python_version >= "3.10" and python_version < "3.11"
pyyaml==6.0.2 ; python_version >= "3.10" and python_version < "3.11"
scikit-learn==1.5.1 ; python_version >= "3.10" and python_version < "3.11"
scipy==1.14.1 ; python_version >= "3.10" and python_version < "3.11"
seaborn==0.13.2 ; python_version >= "3.10" and python_version < "3.11"
setuptools==74.1.2 ; python_version >= "3.10" and python_version < "3.11"
six==1.16.0 ; python_version >= "3.10" and python_version < "3.11"
sympy==1.13.2 ; python_version >= "3.10" and python_version < "3.11"
threadpoolctl==3.5.0 ; python_version >= "3.10" and python_version < "3.11"
torch==2.4.1+cu124 ; python_version >= "3.10" and python_version < "3.11"
torchmetrics==1.4.1 ; python_version >= "3.10" and python_version < "3.11"
tqdm==4.66.5 ; python_version >= "3.10" and python_version < "3.11"
triton==3.0.0 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version < "3.11" and python_version >= "3.10"
typing-extensions==4.12.2 ; python_version >= "3.10" and python_version < "3.11"
tzdata==2024.1 ; python_version >= "3.10" and python_version < "3.11"
yarl==1.10.0 ; python_version >= "3.10" and python_version < "3.11"