forked from microsoft/Megatron-DeepSpeed
-
Notifications
You must be signed in to change notification settings - Fork 0
/
environment.yml
130 lines (130 loc) · 4.09 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
name: labvenv
channels:
- refractor
- <unknown>
- conda-forge
dependencies:
- _libgcc_mutex=0.1=conda_forge
- _openmp_mutex=4.5=2_kmp_llvm
- _sysroot_linux-64_curr_repodata_hack=3=h69a702a_14
- annotated-types=0.6.0=pyhd8ed1ab_0
- binutils_impl_linux-64=2.40=hf600244_0
- blas=2.121=openblas
- blas-devel=3.9.0=21_linux64_openblas
- brotli-python=1.1.0=py38h17151c0_1
- bzip2=1.0.8=hd590300_5
- ca-certificates=2024.2.2=hbcca054_0
- certifi=2024.2.2=pyhd8ed1ab_0
- charset-normalizer=3.3.2=pyhd8ed1ab_0
- click=8.1.7=unix_pyh707e725_0
- colorama=0.4.6=pyhd8ed1ab_0
- deepspeed=0.13.1=cuda120py38hea54ba3_200
- filelock=3.13.1=pyhd8ed1ab_0
- freetype=2.12.1=h267a509_2
- fsspec=2024.2.0=pyhca7485f_0
- gcc=9.5.0=h1fea6ba_13
- gcc_impl_linux-64=9.5.0=h99780fb_19
- gettext=0.21.1=h27087fc_0
- gh=2.43.1=ha8f183a_0
- gmp=6.3.0=h59595ed_0
- gmpy2=2.1.2=py38h793c122_1
- gxx=9.5.0=h1fea6ba_13
- gxx_impl_linux-64=9.5.0=h99780fb_19
- hjson-py=3.1.0=pyhd8ed1ab_0
- icu=73.2=h59595ed_0
- idna=3.6=pyhd8ed1ab_0
- jinja2=3.1.3=pyhd8ed1ab_0
- joblib=1.3.2=pyhd8ed1ab_0
- kernel-headers_linux-64=3.10.0=h4a8ded7_14
- lcms2=2.16=hb7c19ff_0
- ld_impl_linux-64=2.40=h41732ed_0
- lerc=4.0.0=h27087fc_0
- libaio=0.3.113=h166bdaf_0
- libblas=3.9.0=21_linux64_openblas
- libcblas=3.9.0=21_linux64_openblas
- libdeflate=1.19=hd590300_0
- libffi=3.4.2=h7f98852_5
- libgcc-devel_linux-64=9.5.0=h0a57e50_19
- libgcc-ng=13.2.0=h807b86a_5
- libgcrypt=1.10.3=hd590300_0
- libgfortran-ng=13.2.0=h69a702a_5
- libgfortran5=13.2.0=ha4646dd_5
- libgomp=13.2.0=h807b86a_5
- libgpg-error=1.48=h71f35ed_0
- libhwloc=2.9.3=default_h554bfaf_1009
- libibverbs-cos7-x86_64=22.4=h9b0a68f_1105
- libiconv=1.17=hd590300_2
- libjpeg-turbo=3.0.0=hd590300_1
- liblapack=3.9.0=21_linux64_openblas
- liblapacke=3.9.0=21_linux64_openblas
- libnsl=2.0.1=hd590300_0
- libopenblas=0.3.26=pthreads_h413a1c8_0
- libpng=1.6.43=h2797004_0
- libsanitizer=9.5.0=h2f262e1_19
- libsqlite=3.45.1=h2797004_0
- libstdcxx-devel_linux-64=9.5.0=h0a57e50_19
- libstdcxx-ng=13.2.0=h7e041cc_5
- libtiff=4.6.0=ha9c0a0a_2
- libuuid=2.38.1=h0b41bf4_0
- libwebp-base=1.3.2=hd590300_0
- libxcb=1.15=h0b41bf4_0
- libxcrypt=4.4.36=hd590300_1
- libxml2=2.12.5=h232c23b_0
- libzlib=1.2.13=hd590300_5
- llvm-openmp=17.0.6=h4dfa4b3_0
- markupsafe=2.1.5=py38h01eb140_0
- mkl=2024.0.0=ha957f24_49657
- mkl-devel=2024.0.0=ha770c72_49657
- mkl-include=2024.0.0=ha957f24_49657
- mpc=1.3.1=hfe3b2da_0
- mpfr=4.2.1=h9458935_0
- mpmath=1.3.0=pyhd8ed1ab_0
- ncurses=6.4=h59595ed_2
- networkx=3.1=pyhd8ed1ab_0
- ninja-build=1.9.0=h6bb024c_0
- nltk=3.8.1=pyhd8ed1ab_0
- numpy=1.24.4=py38h59b608b_0
- nvtx=0.2.10=py38h01eb140_0
- openblas=0.3.26=pthreads_h7a3da1a_0
- openjpeg=2.5.2=h488ebb8_0
- openssl=3.2.1=hd590300_0
- packaging=23.2=pyhd8ed1ab_0
- pillow=10.2.0=py38ha43c96d_0
- pip=24.0=pyhd8ed1ab_0
- psutil=5.9.8=py38h01eb140_0
- pthread-stubs=0.4=h36c2ea0_1001
- py-cpuinfo=9.0.0=pyhd8ed1ab_0
- pybind11=2.11.1=py38h7f3f72f_2
- pybind11-global=2.11.1=py38h7f3f72f_2
- pydantic=2.6.3=pyhd8ed1ab_0
- pydantic-core=2.16.3=py38h0cc4f7c_0
- pynvml=11.5.0=pyhd8ed1ab_0
- pysocks=1.7.1=pyha2e5f31_6
- python=3.8.18=hd12c33a_1_cpython
- python_abi=3.8=4_cp38
- pytorch=2.1.0=py3.8_cuda12.1_cudnn8.9.2_0
- pyyaml=6.0.1=py38h01eb140_1
- rdma-core-cos7-x86_64=22.4=h9b0a68f_1105
- readline=8.2=h8228510_1
- regex=2023.12.25=py38h01eb140_0
- requests=2.31.0=pyhd8ed1ab_0
- setuptools=69.1.1=pyhd8ed1ab_0
- six=1.16.0=pyh6c4a22f_0
- sympy=1.12=pypyh9d50eac_103
- sysroot_linux-64=2.17=h4a8ded7_14
- tbb=2021.11.0=h00ab1b0_1
- tk=8.6.13=noxft_h4845f30_101
- torchvision=0.16.0=py38_cu121
- tqdm=4.66.2=pyhd8ed1ab_0
- typing-extensions=4.10.0=hd8ed1ab_0
- typing_extensions=4.10.0=pyha770c72_0
- urllib3=2.2.1=pyhd8ed1ab_0
- wheel=0.42.0=pyhd8ed1ab_0
- xorg-libxau=1.0.11=hd590300_0
- xorg-libxdmcp=1.1.3=h7f98852_0
- xz=5.2.6=h166bdaf_0
- yaml=0.2.5=h7f98852_2
- zstd=1.5.5=hfc55251_0
- pip:
- apex==0.1
- ninja==1.11.1.1