-
Notifications
You must be signed in to change notification settings - Fork 0
/
environment.yaml
211 lines (211 loc) · 5.73 KB
/
environment.yaml
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
199
200
201
202
203
204
205
206
207
208
209
210
211
name: uldenv
channels:
- pytorch
- nvidia
- nvidia/label/cuda-11.8.0
- defaults
dependencies:
- _libgcc_mutex=0.1=main
- _openmp_mutex=5.1=1_gnu
- blas=1.0=mkl
- brotli-python=1.0.9=py310h6a678d5_8
- bzip2=1.0.8=h5eee18b_6
- ca-certificates=2024.3.11=h06a4308_0
- certifi=2024.6.2=py310h06a4308_0
- charset-normalizer=2.0.4=pyhd3eb1b0_0
- cuda-cccl=11.8.89=0
- cuda-command-line-tools=11.8.0=0
- cuda-compiler=11.8.0=0
- cuda-cudart=11.8.89=0
- cuda-cudart-dev=11.8.89=0
- cuda-cuobjdump=11.8.86=0
- cuda-cupti=11.8.87=0
- cuda-cuxxfilt=11.8.86=0
- cuda-documentation=11.8.86=0
- cuda-driver-dev=11.8.89=0
- cuda-gdb=11.8.86=0
- cuda-libraries=11.8.0=0
- cuda-libraries-dev=11.8.0=0
- cuda-memcheck=11.8.86=0
- cuda-nsight=11.8.86=0
- cuda-nsight-compute=11.8.0=0
- cuda-nvcc=11.8.89=0
- cuda-nvdisasm=11.8.86=0
- cuda-nvml-dev=11.8.86=0
- cuda-nvprof=11.8.87=0
- cuda-nvprune=11.8.86=0
- cuda-nvrtc=11.8.89=0
- cuda-nvrtc-dev=11.8.89=0
- cuda-nvtx=11.8.86=0
- cuda-nvvp=11.8.87=0
- cuda-profiler-api=11.8.86=0
- cuda-runtime=11.8.0=0
- cuda-sanitizer-api=11.8.86=0
- cuda-toolkit=11.8.0=0
- cuda-tools=11.8.0=0
- cuda-version=12.5=3
- cuda-visual-tools=11.8.0=0
- ffmpeg=4.3=hf484d3e_0
- filelock=3.13.1=py310h06a4308_0
- freetype=2.12.1=h4a9f257_0
- gds-tools=1.4.0.31=0
- gmp=6.2.1=h295c915_3
- gmpy2=2.1.2=py310heeb90bb_0
- gnutls=3.6.15=he1e5248_0
- idna=3.7=py310h06a4308_0
- intel-openmp=2023.1.0=hdb19cb5_46306
- jinja2=3.1.4=py310h06a4308_0
- jpeg=9e=h5eee18b_1
- lame=3.100=h7b6447c_0
- lcms2=2.12=h3be6417_0
- ld_impl_linux-64=2.38=h1181459_1
- lerc=3.0=h295c915_0
- libcublas=11.11.3.6=0
- libcublas-dev=11.11.3.6=0
- libcufft=10.9.0.58=0
- libcufft-dev=10.9.0.58=0
- libcufile=1.10.0.4=0
- libcufile-dev=1.4.0.31=0
- libcurand=10.3.6.39=0
- libcurand-dev=10.3.0.86=0
- libcusolver=11.4.1.48=0
- libcusolver-dev=11.4.1.48=0
- libcusparse=11.7.5.86=0
- libcusparse-dev=11.7.5.86=0
- libdeflate=1.17=h5eee18b_1
- libffi=3.4.4=h6a678d5_1
- libgcc-ng=11.2.0=h1234567_1
- libgomp=11.2.0=h1234567_1
- libiconv=1.16=h5eee18b_3
- libidn2=2.3.4=h5eee18b_0
- libjpeg-turbo=2.0.0=h9bf148f_0
- libnpp=11.8.0.86=0
- libnpp-dev=11.8.0.86=0
- libnvjpeg=11.9.0.86=0
- libnvjpeg-dev=11.9.0.86=0
- libpng=1.6.39=h5eee18b_0
- libstdcxx-ng=11.2.0=h1234567_1
- libtasn1=4.19.0=h5eee18b_0
- 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=py310h5eee18b_0
- mkl=2023.1.0=h213fc3f_46344
- mkl-service=2.4.0=py310h5eee18b_1
- mkl_fft=1.3.8=py310h5eee18b_0
- mkl_random=1.2.4=py310hdb19cb5_0
- mpc=1.1.0=h10f8cd9_1
- mpfr=4.0.2=hb69a4c5_1
- mpmath=1.3.0=py310h06a4308_0
- ncurses=6.4=h6a678d5_0
- nettle=3.7.3=hbbd107a_1
- networkx=3.2.1=py310h06a4308_0
- nsight-compute=2022.3.0.22=0
- numpy=1.26.4=py310h5f9d8c6_0
- numpy-base=1.26.4=py310hb5e798b_0
- openh264=2.1.1=h4ff587b_0
- openjpeg=2.4.0=h3ad879b_0
- openssl=3.0.13=h7f8727e_2
- pillow=10.3.0=py310h5eee18b_0
- pip=24.0=py310h06a4308_0
- pysocks=1.7.1=py310h06a4308_0
- python=3.10.14=h955ad1f_1
- pytorch=2.1.1=py3.10_cuda11.8_cudnn8.7.0_0
- pytorch-cuda=11.8=h7e8668a_5
- pytorch-mutex=1.0=cuda
- pyyaml=6.0.1=py310h5eee18b_0
- readline=8.2=h5eee18b_0
- requests=2.32.2=py310h06a4308_0
- setuptools=69.5.1=py310h06a4308_0
- sqlite=3.45.3=h5eee18b_0
- sympy=1.12=py310h06a4308_0
- tbb=2021.8.0=hdb19cb5_0
- tk=8.6.14=h39e8969_0
- torchaudio=2.1.1=py310_cu118
- torchtriton=2.1.0=py310
- torchvision=0.16.1=py310_cu118
- typing_extensions=4.11.0=py310h06a4308_0
- urllib3=2.2.1=py310h06a4308_0
- wheel=0.43.0=py310h06a4308_0
- xz=5.4.6=h5eee18b_1
- yaml=0.2.5=h7b6447c_0
- zlib=1.2.13=h5eee18b_1
- zstd=1.5.5=hc292b87_2
- pip:
- absl-py==2.1.0
- accelerate==0.31.0
- aiohttp==3.9.5
- aiosignal==1.3.1
- annotated-types==0.7.0
- antlr4-python3-runtime==4.9.3
- async-timeout==4.0.3
- attrs==23.2.0
- bitsandbytes==0.43.1
- click==8.1.7
- codetiming==1.4.0
- colorama==0.4.6
- datasets==2.17.1
- deepspeed==0.14.2
- dill==0.3.8
- docker-pycreds==0.4.0
- einops==0.8.0
- evaluate==0.4.2
- flash-attn==2.5.6
- frozenlist==1.4.1
- fsspec==2023.10.0
- gitdb==4.0.11
- gitpython==3.1.43
- hjson==3.1.0
- huggingface-hub==0.23.3
- hydra-core==1.3.2
- joblib==1.4.2
- lightning==2.2.1
- lightning-utilities==0.11.2
- lxml==5.2.2
- markdown-it-py==3.0.0
- mdurl==0.1.2
- multidict==6.0.5
- multiprocess==0.70.16
- ninja==1.11.1.1
- nltk==3.8.1
- omegaconf==2.3.0
- packaging==24.1
- pandas==2.2.2
- peft==0.11.1
- platformdirs==4.2.2
- portalocker==2.8.2
- protobuf==5.27.1
- psutil==5.9.8
- py-cpuinfo==9.0.0
- pyarrow==16.1.0
- pyarrow-hotfix==0.6
- pydantic==2.7.3
- pydantic-core==2.18.4
- pygments==2.18.0
- pynvml==11.5.0
- python-dateutil==2.9.0.post0
- pytorch-lightning==2.2.1
- pytz==2024.1
- regex==2024.5.15
- rich==13.7.1
- rouge-score==0.1.2
- sacrebleu==2.4.2
- safetensors==0.4.3
- scipy==1.13.1
- sentry-sdk==2.5.1
- setproctitle==1.3.3
- six==1.16.0
- smmap==5.0.1
- structlog==24.2.0
- tabulate==0.9.0
- tokenizers==0.15.2
- torchmetrics==1.4.0.post0
- tqdm==4.66.4
- transformers==4.38.1
- tzdata==2024.1
- wandb==0.17.1
- xxhash==3.4.1
- yarl==1.9.4