forked from soarsmu/SVD-Bench
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathenvironment.yml
198 lines (198 loc) · 5.81 KB
/
environment.yml
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
name: svd_bench
channels:
- pytorch
- huggingface
- nvidia
- conda-forge
- defaults
dependencies:
- _libgcc_mutex=0.1=conda_forge
- _openmp_mutex=4.5=2_gnu
- accelerate=0.33.0=pyhd8ed1ab_0
- aiohappyeyeballs=2.3.6=pyhd8ed1ab_0
- aiohttp=3.10.4=py311h61187de_0
- aiosignal=1.3.1=pyhd8ed1ab_0
- arrow-cpp=16.1.0=hc1eb8f0_0
- attrs=24.2.0=pyh71513ae_0
- aws-c-auth=0.6.19=h5eee18b_0
- aws-c-cal=0.5.20=hff2c3d7_3
- aws-c-common=0.8.5=h166bdaf_0
- aws-c-compression=0.2.16=hf5f93bc_0
- aws-c-event-stream=0.2.15=h6a678d5_0
- aws-c-http=0.6.25=h5eee18b_0
- aws-c-io=0.13.10=h2666983_0
- aws-c-mqtt=0.7.13=h5eee18b_0
- aws-c-s3=0.1.51=hdbd6064_0
- aws-c-sdkutils=0.1.6=hf5f93bc_1
- aws-checksums=0.1.13=hf5f93bc_5
- aws-crt-cpp=0.18.16=h6a678d5_0
- aws-sdk-cpp=1.10.55=h721c034_0
- blas=1.0=mkl
- boost-cpp=1.78.0=he72f1d9_0
- bottleneck=1.4.0=py311h18e1886_1
- brotli-python=1.0.9=py311h6a678d5_8
- bzip2=1.0.8=h5eee18b_6
- c-ares=1.33.0=ha66036c_0
- ca-certificates=2024.7.4=hbcca054_0
- certifi=2024.7.4=pyhd8ed1ab_0
- charset-normalizer=3.3.2=pyhd3eb1b0_0
- colorama=0.4.6=pyhd8ed1ab_0
- cuda-cudart=12.1.105=0
- cuda-cupti=12.1.105=0
- cuda-libraries=12.1.0=0
- cuda-nvrtc=12.1.105=0
- cuda-nvtx=12.1.105=0
- cuda-opencl=12.6.37=0
- cuda-runtime=12.1.0=0
- cuda-version=12.6=3
- dataclasses=0.8=pyhc8e2a94_3
- datasets=2.21.0=py_0
- dill=0.3.8=pyhd8ed1ab_0
- ffmpeg=4.3=hf484d3e_0
- filelock=3.13.1=py311h06a4308_0
- freetype=2.12.1=h4a9f257_0
- frozenlist=1.4.1=py311h459d7ec_0
- fsspec=2024.6.1=pyhff2d567_0
- gflags=2.2.2=he1b5a44_1004
- glog=0.5.0=h48cff8f_0
- gmp=6.2.1=h295c915_3
- gmpy2=2.1.2=py311hc9b5ff0_0
- gnutls=3.6.15=he1e5248_0
- huggingface_hub=0.24.5=py_0
- icu=70.1=h27087fc_0
- idna=3.7=py311h06a4308_0
- intel-openmp=2023.1.0=hdb19cb5_46306
- jinja2=3.1.4=py311h06a4308_0
- jpeg=9e=h5eee18b_3
- krb5=1.20.1=h143b758_1
- lame=3.100=h7b6447c_0
- lcms2=2.12=h3be6417_0
- ld_impl_linux-64=2.38=h1181459_1
- lerc=3.0=h295c915_0
- libabseil=20240116.2=cxx17_h6a678d5_0
- libbrotlicommon=1.0.9=h166bdaf_9
- libbrotlidec=1.0.9=h166bdaf_9
- libbrotlienc=1.0.9=h166bdaf_9
- libcublas=12.1.0.26=0
- libcufft=11.0.2.4=0
- libcufile=1.11.0.15=0
- libcurand=10.3.7.37=0
- libcurl=8.7.1=h251f7ec_0
- libcusolver=11.4.4.55=0
- libcusparse=12.0.2.55=0
- libdeflate=1.17=h5eee18b_1
- libedit=3.1.20230828=h5eee18b_0
- libev=4.33=hd590300_2
- libevent=2.1.12=hf998b51_1
- libffi=3.4.4=h6a678d5_1
- libgcc-ng=14.1.0=h77fa898_0
- libgomp=14.1.0=h77fa898_0
- libgrpc=1.62.2=h2d74bed_0
- libiconv=1.16=h5eee18b_3
- libidn2=2.3.4=h5eee18b_0
- libjpeg-turbo=2.0.0=h9bf148f_0
- libnghttp2=1.57.0=h2d74bed_0
- libnpp=12.0.2.50=0
- libnvjitlink=12.1.105=0
- libnvjpeg=12.1.1.14=0
- libpng=1.6.39=h5eee18b_0
- libprotobuf=4.25.3=he621ea3_0
- libssh2=1.10.0=ha35d2d1_2
- libstdcxx-ng=11.2.0=h1234567_1
- libtasn1=4.19.0=h5eee18b_0
- libthrift=0.15.0=h1795dd8_2
- libtiff=4.5.1=h6a678d5_0
- libunistring=0.9.10=h27cfd23_0
- libuuid=1.41.5=h5eee18b_0
- libwebp-base=1.3.2=h5eee18b_0
- llvm-openmp=14.0.6=h9e868ea_0
- lz4-c=1.9.4=h6a678d5_1
- markupsafe=2.1.3=py311h5eee18b_0
- mkl=2023.1.0=h213fc3f_46344
- mkl-service=2.4.0=py311h5eee18b_1
- mkl_fft=1.3.8=py311h5eee18b_0
- mkl_random=1.2.4=py311hdb19cb5_0
- mpc=1.1.0=h10f8cd9_1
- mpfr=4.0.2=hb69a4c5_1
- mpmath=1.3.0=py311h06a4308_0
- multidict=6.0.5=py311h459d7ec_0
- multiprocess=0.70.16=py311h459d7ec_0
- ncurses=6.4=h6a678d5_0
- nettle=3.7.3=hbbd107a_1
- networkx=3.3=py311h06a4308_0
- numexpr=2.8.7=py311h65dcdc2_0
- numpy=1.26.4=py311h08b1b3b_0
- numpy-base=1.26.4=py311hf175353_0
- openh264=2.1.1=h4ff587b_0
- openjpeg=2.5.2=he7f1fd0_0
- openssl=3.3.1=h4bc722e_2
- orc=2.0.1=h2d29ad5_0
- packaging=24.1=pyhd8ed1ab_0
- pandas=2.2.2=py311ha02d727_0
- pillow=10.4.0=py311h5eee18b_0
- pip=24.2=py311h06a4308_0
- psutil=6.0.0=py311h331c9d8_0
- pyarrow=16.1.0=py311ha02d727_0
- pyarrow-hotfix=0.6=pyhd8ed1ab_0
- pysocks=1.7.1=py311h06a4308_0
- python=3.11.9=h955ad1f_0
- python-dateutil=2.9.0=pyhd8ed1ab_0
- python-tzdata=2024.1=pyhd8ed1ab_0
- python-xxhash=3.4.1=py311h459d7ec_0
- python_abi=3.11=2_cp311
- pytorch=2.1.2=py3.11_cuda12.1_cudnn8.9.2_0
- pytorch-cuda=12.1=ha16c6d3_5
- pytorch-mutex=1.0=cuda
- pytz=2024.1=pyhd8ed1ab_0
- pyyaml=6.0.1=py311h5eee18b_0
- re2=2022.04.01=h27087fc_0
- readline=8.2=h5eee18b_0
- regex=2024.7.24=py311h61187de_0
- requests=2.32.3=py311h06a4308_0
- s2n=1.3.27=h1b5e8cb_0
- safetensors=0.4.4=py311hb3a8bbb_0
- setuptools=72.1.0=py311h06a4308_0
- six=1.16.0=pyh6c4a22f_0
- snappy=1.2.1=h6a678d5_0
- sqlite=3.45.3=h5eee18b_0
- sympy=1.12=py311h06a4308_0
- tbb=2021.8.0=hdb19cb5_0
- tk=8.6.14=h39e8969_0
- tokenizers=0.19.1=py311h585d796_0
- torchaudio=2.1.2=py311_cu121
- torchtriton=2.1.0=py311
- torchvision=0.16.2=py311_cu121
- tqdm=4.66.5=pyhd8ed1ab_0
- transformers=4.44.0=pyhd8ed1ab_0
- typing-extensions=4.11.0=py311h06a4308_0
- typing_extensions=4.11.0=py311h06a4308_0
- tzdata=2024a=h04d1e81_0
- urllib3=2.2.2=py311h06a4308_0
- utf8proc=2.6.1=h5eee18b_1
- wheel=0.43.0=py311h06a4308_0
- xxhash=0.8.2=hd590300_0
- xz=5.4.6=h5eee18b_1
- yaml=0.2.5=h7b6447c_0
- yarl=1.9.4=py311h459d7ec_0
- zlib=1.2.13=h5eee18b_1
- zstd=1.5.5=hc292b87_2
- pip:
- bitsandbytes==0.43.3
- docstring-parser==0.16
- einops==0.8.0
- flash-attn==2.6.3
- huggingface-hub==0.24.5
- joblib==1.4.2
- markdown-it-py==3.0.0
- mdurl==0.1.2
- peft==0.12.0
- prettytable==3.11.0
- pygments==2.18.0
- rich==13.7.1
- scikit-learn==1.5.1
- scipy==1.14.0
- shtab==1.7.1
- threadpoolctl==3.5.0
- trl==0.9.6
- tyro==0.8.8
- wcwidth==0.2.13