-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrequirements.txt
More file actions
56 lines (50 loc) · 1.58 KB
/
requirements.txt
File metadata and controls
56 lines (50 loc) · 1.58 KB
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
# PIP-Requirements for Benchmark (2025)
# pip install -r requirements.txt
# hint: use env
# python3 -m venv env_elra
# source env_elra/bin/activate
certifi>=2025.6.15
requests>=2.32.3
# numpy~=1.26.4 # torch 2.2.2 cannot use numpy=2
numpy~=2.1.3 # torch 2.4.1 + numpy 2.1.1
scipy>=1.14.1 # no direct import
#six>=1.16.0 # no direct import
#sympy>=1.13.3 # no direct import
#mmh3~=4.1.0 # hash128
timm>=1.0.16 # 0.9.12, 1.0.11
#scikit-learn~=1.5.2
#mpmath>=1.3.0
#matplotlib>=3.8.1
#tabulate>=0.9.0 # no direct import (torch compile)
#fastparquet>=2024.5.0 # optional (disabled) + issue_p312
#Pillow>=10.4.0 # no direct import
# cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR: cudnnFinalize Descriptor Failed cudnn_status: CUDNN_STATUS_NOT_SUPPORTED
# torch=2.3 failed with float16 and Windows-MKL-issue
torch~=2.7.0 # 2.6 avial+working
torchaudio~=2.7.0
torchvision~=0.22.0 # 2.4=0.19.0
#torchvideo>=0.0.0
#seaborn>=0.13.2 # unused (statistical data visualization)
#pyarrow>=14.0.1 # only for FunctionBenchmark.py
pandas>=2.1.2 # NNBenchmark.py: optional (disabled)
#Jinja2>=3.1.2 # unused
#resnet>=0.1 # unused
#snakeviz>=2.2.0
#Cython-3.0.2
#numba>=0.58.0
#keras>=2.15.0
lion-pytorch>=0.2.3
dog-optimizer>=1.0.3 # optional
#torch-optimizer>=0.3.0 # optional
#typing>=3.7.4.3 # optional (for Sophia)
# cffi>=1.17.0 # optional
#imageio>=2.33.1
#typing-extensions>=4.9.0 # torch-2 on Python 3.9
#datasets>=2.16.1
#deeplake-3.8.11
# CPU/GPU/etc
#pip3 install torch torchvision torchaudio
#torch==2.1.0.dev20230629+cpu
#torchaudio==2.1.0.dev20230629+cpu
#torchvision==0.16.0.dev20230629+cpu
#EOF.