A docker environment for AudioCraft with GPU/CUDA.
- Install NVIDIA driver for CUDA 11.8 and setup Docker enviroment on your host machine, then check CUDA version.
nvidia-smi
- Build and run docker by compose.
docker compose up -d --build
- Open bash in docker.
docker compose exec app bash
- Run an AudioGen sample.
poetry run python src/main.py