| # Single GPU | |
| # python scripts/train_voxcpm_finetune.py --config_path conf/librispeech_full_voxcpm_v1_whisper.yaml | |
| # Multi-GPU (2×) | |
| source ./.venv/bin/activate | |
| export TORCH_NCCL_HEARTBEAT_TIMEOUT_SEC="${TORCH_NCCL_HEARTBEAT_TIMEOUT_SEC:-3600}" | |
| # print python env info | |
| echo "Python executable: $(which python)" | |
| CUDA_VISIBLE_DEVICES=0,1 torchrun --nproc_per_node=2 \ | |
| scripts/train_voxcpm_finetune.py --config_path conf/librispeech_full_voxcpm_v1_whisper.yaml | |