Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
environment.yaml 4.28 KB
Newer Older
Rayyyyy's avatar
First add.  
Rayyyyy committed
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 
name: llm-embedder
channels:
  - pytorch
  - nvidia
  - conda-forge
  - defaults
dependencies:
  - _libgcc_mutex=0.1=main
  - _openmp_mutex=5.1=1_gnu
  - blas=1.0=mkl
  - bzip2=1.0.8=h7b6447c_0
  - ca-certificates=2023.7.22=hbcca054_0
  - cuda-cudart=11.8.89=0
  - cuda-cupti=11.8.87=0
  - cuda-libraries=11.8.0=0
  - cuda-nvrtc=11.8.89=0
  - cuda-nvtx=11.8.86=0
  - cuda-runtime=11.8.0=0
  - cudatoolkit=11.8.0=h6a678d5_0
  - faiss=1.7.2=py310cuda112h76f6547_0_cuda
  - faiss-gpu=1.7.2=h788eb59_4
  - filelock=3.9.0=py310h06a4308_0
  - gmp=6.2.1=h295c915_3
  - gmpy2=2.1.2=py310heeb90bb_0
  - intel-openmp=2021.4.0=h06a4308_3561
  - jinja2=3.1.2=py310h06a4308_0
  - ld_impl_linux-64=2.38=h1181459_1
  - libblas=3.9.0=12_linux64_mkl
  - libcblas=3.9.0=12_linux64_mkl
  - libcublas=11.11.3.6=0
  - libcufft=10.9.0.58=0
  - libcufile=1.8.0.34=0
  - libcurand=10.3.4.52=0
  - libcusolver=11.4.1.48=0
  - libcusparse=11.7.5.86=0
  - libfaiss=1.7.2=cuda112hc9ed507_0_cuda
  - libfaiss-avx2=1.7.2=cuda112h1234567_0_cuda
  - libffi=3.4.4=h6a678d5_0
  - libgcc-ng=11.2.0=h1234567_1
  - libgomp=11.2.0=h1234567_1
  - liblapack=3.9.0=12_linux64_mkl
  - libnpp=11.8.0.86=0
  - libnvjpeg=11.9.0.86=0
  - libstdcxx-ng=11.2.0=h1234567_1
  - libuuid=1.41.5=h5eee18b_0
  - llvm-openmp=14.0.6=h9e868ea_0
  - markupsafe=2.1.1=py310h7f8727e_0
  - mkl=2021.4.0=h06a4308_640
  - mpc=1.1.0=h10f8cd9_1
  - mpfr=4.0.2=hb69a4c5_1
  - mpmath=1.3.0=py310h06a4308_0
  - ncurses=6.4=h6a678d5_0
  - networkx=3.1=py310h06a4308_0
  - openssl=3.0.11=h7f8727e_2
  - pip=23.3=py310h06a4308_0
  - python=3.10.13=h955ad1f_0
  - python_abi=3.10=2_cp310
  - pytorch=2.1.0=py3.10_cuda11.8_cudnn8.7.0_0
  - pytorch-cuda=11.8=h7e8668a_5
  - pytorch-mutex=1.0=cuda
  - pyyaml=6.0=py310h5eee18b_1
  - readline=8.2=h5eee18b_0
  - setuptools=68.0.0=py310h06a4308_0
  - sqlite=3.41.2=h5eee18b_0
  - sympy=1.11.1=py310h06a4308_0
  - tk=8.6.12=h1ccaba5_0
  - torchtriton=2.1.0=py310
  - typing_extensions=4.7.1=py310h06a4308_0
  - wheel=0.41.2=py310h06a4308_0
  - xz=5.4.2=h5eee18b_0
  - yaml=0.2.5=h7b6447c_0
  - zlib=1.2.13=h5eee18b_0
  - pip:
      - accelerate==0.23.0
      - aiohttp==3.8.6
      - aiosignal==1.3.1
      - asttokens==2.4.0
      - async-timeout==4.0.3
      - attrs==23.1.0
      - backcall==0.2.0
      - cachetools==5.3.1
      - certifi==2023.7.22
      - charset-normalizer==3.3.0
      - click==8.1.7
      - comm==0.1.4
      - contourpy==1.1.1
      - cycler==0.12.1
      - datasets==2.14.5
      - debugpy==1.8.0
      - decorator==5.1.1
      - deepspeed==0.11.1
      - dill==0.3.7
      - exceptiongroup==1.1.3
      - executing==2.0.0
      - fonttools==4.43.1
      - frozenlist==1.4.0
      - fsspec==2023.6.0
      - hjson==3.1.0
      - huggingface-hub==0.17.3
      - idna==3.4
      - ipykernel==6.25.2
      - ipython==8.16.1
      - ipywidgets==8.1.1
      - jedi==0.19.1
      - joblib==1.3.2
      - jupyter-client==8.4.0
      - jupyter-core==5.4.0
      - jupyterlab-widgets==3.0.9
      - kiwisolver==1.4.5
      - matplotlib==3.8.0
      - matplotlib-inline==0.1.6
      - multidict==6.0.4
      - multiprocess==0.70.15
      - nest-asyncio==1.5.8
      - ninja==1.11.1.1
      - nltk==3.8.1
      - numpy==1.26.1
      - nvidia-ml-py==12.535.108
      - nvitop==1.3.1
      - packaging==23.2
      - pandas==2.1.1
      - parso==0.8.3
      - pexpect==4.8.0
      - pickleshare==0.7.5
      - pillow==10.1.0
      - platformdirs==3.11.0
      - prompt-toolkit==3.0.39
      - psutil==5.9.6
      - ptyprocess==0.7.0
      - pure-eval==0.2.2
      - py-cpuinfo==9.0.0
      - pyarrow==13.0.0
      - pydantic==1.10.13
      - pygments==2.16.1
      - pyparsing==3.1.1
      - python-dateutil==2.8.2
      - pytz==2023.3.post1
      - pyzmq==25.1.1
      - regex==2023.10.3
      - requests==2.31.0
      - rouge==1.0.1
      - safetensors==0.4.0
      - scikit-learn==1.3.1
      - scipy==1.11.3
      - seaborn==0.13.0
      - sentence-transformers==2.2.2
      - sentencepiece==0.1.99
      - six==1.16.0
      - stack-data==0.6.3
      - termcolor==2.3.0
      - threadpoolctl==3.2.0
      - tokenizers==0.14.1
      - torchvision==0.16.0
      - tornado==6.3.3
      - tqdm==4.66.1
      - traitlets==5.11.2
      - transformers==4.34.1
      - tzdata==2023.3
      - urllib3==2.0.7
      - wcwidth==0.2.8
      - widgetsnbextension==4.0.9
      - xxhash==3.4.1
      - yarl==1.9.2