environment.yaml

name: mamba
channels:
  - nvidia/label/cuda-11.8.0
  - defaults
dependencies:
  - _libgcc_mutex=0.1=main
  - _openmp_mutex=5.1=1_gnu
  - bzip2=1.0.8=h5eee18b_6
  - ca-certificates=2024.9.24=h06a4308_0
  - cuda-nvcc=11.8.89=0
  - cudatoolkit=11.8.0=h6a678d5_0
  - ld_impl_linux-64=2.40=h12ee557_0
  - libffi=3.4.4=h6a678d5_1
  - libgcc-ng=11.2.0=h1234567_1
  - libgomp=11.2.0=h1234567_1
  - libstdcxx-ng=11.2.0=h1234567_1
  - libuuid=1.41.5=h5eee18b_0
  - ncurses=6.4=h6a678d5_0
  - openssl=3.0.15=h5eee18b_0
  - packaging=24.1=py310h06a4308_0
  - pip=24.2=py310h06a4308_0
  - python=3.10.15=he870216_1
  - readline=8.2=h5eee18b_0
  - setuptools=75.1.0=py310h06a4308_0
  - sqlite=3.45.3=h5eee18b_0
  - tk=8.6.14=h39e8969_0
  - wheel=0.44.0=py310h06a4308_0
  - xz=5.4.6=h5eee18b_1
  - zlib=1.2.13=h5eee18b_1
  - pip:
      - absl-py==2.1.0
      - attrs==24.2.0
      - causal-conv1d==1.4.0
      - certifi==2022.12.7
      - charset-normalizer==2.1.1
      - colorama==0.4.4
      - colorlog==4.7.2
      - einops==0.8.0
      - filelock==3.13.1
      - fsspec==2024.2.0
      - grpcio==1.68.0
      - huggingface-hub==0.26.2
      - idna==3.4
      - jinja2==3.1.3
      - joblib==1.4.2
      - jsonlines==4.0.0
      - kmeans-pytorch==0.3
      - mamba-ssm==2.2.2
      - markdown==3.7
      - markupsafe==2.1.5
      - mpmath==1.3.0
      - networkx==3.2.1
      - ninja==1.11.1.2
      - numpy==1.26.3
      - pandas==2.2.3
      - pillow==10.2.0
      - plotly==5.24.1
      - protobuf==5.29.0
      - psutil==6.1.0
      - python-dateutil==2.9.0.post0
      - pytz==2024.2
      - pyyaml==6.0.2
      - recbole==1.2.0
      - regex==2024.11.6
      - requests==2.28.1
      - safetensors==0.4.5
      - scikit-learn==1.5.2
      - scipy==1.14.1
      - six==1.16.0
      - sympy==1.13.1
      - tabulate==0.9.0
      - tenacity==9.0.0
      - tensorboard==2.18.0
      - tensorboard-data-server==0.7.2
      - texttable==1.7.0
      - thop==0.1.1-2209072238
      - threadpoolctl==3.5.0
      - tokenizers==0.20.3
      - torch==2.1.1+cu118
      - torchaudio==2.1.1+cu118
      - torchvision==0.16.1+cu118
      - tqdm==4.67.1
      - transformers==4.46.3
      - triton==2.1.0
      - typing-extensions==4.9.0
      - tzdata==2024.2
      - urllib3==1.26.13
      - werkzeug==3.1.3
prefix: /home/yifengzhang/miniconda3/envs/mamba