Skip to content

lz-hust/DSE-CBM

Repository files navigation

Installation Guide

This guide provides step-by-step instructions to set up the DSE-CBM (a speaker diarization toolkit driven by Pyannote(https://github.com/pyannote/pyannote-audio)) project in a Conda environment on a Linux system with CUDA support.

Installation Steps

1. Update Conda

Update Conda to ensure you have the latest version:

conda update -n base -c defaults conda

2. Create and Activate Conda Environment

Create a new Conda environment named dsecbm with Python 3.12:

conda create -n dsecbm python=3.12
conda activate dsecbm

3. Set Environment Variables

Configure environment variables for CUDA 11.8 by adding the following lines to ~/.bashrc:

export CUDA_HOME="/usr/local/cuda-11.8/"
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH

Apply the changes:

source ~/.bashrc

4. Install PyTorch with CUDA Support

Install CUDA Toolkit 11.8 and PyTorch with its dependencies:

conda install cudatoolkit=11.8 -c nvidia
pip install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu118

5. Install Mamba-SSM

git clone https://github.com/state-spaces/mamba.git
cd mamba/
pip install .

6. Install DSE-CBM

git clone https://github.com/lz-hust/DSE-CBM.git
cd DSE-CBM/
pip install -r requirements.txt

License

  • The code in this repository is licensed under the MIT license.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

Packages

Contributors

Languages