164 lines
4.3 KiB
YAML
164 lines
4.3 KiB
YAML
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
|