AutoPrepSong/set_docker.sh at main · MSLDCherryPick/AutoPrepSong · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29


# Evaluation
# name="AutoPrep"
name="AutoPrepSongV2"
# container="pytorch/pytorch:2.4.0-cuda12.4-cudnn9-runtime"
# name=tmp
sudo docker stop $name
sudo docker rm $name
# container="pytorch/pytorch:2.4.0-cuda12.4-cudnn9-runtime"
# container="nvcr.io/nvidia/pytorch:24.07-py3"
# container="nvcr.io/nvidia/pytorch:24.10-py3"
# container="nvcr.io/nvidia/pytorch:25.11-py3"
# container="nvcr.io/nvidia/pytorch:25.11-py3-amlt-AutoPrepSongV2"
container="nvcr.io/nvidia/pytorch:25.11-py3-amlt-AutoPrepSongV2-diarize"

sudo docker run -d --privileged  --name $name \
    --net=host  --ipc=host  --shm-size 900g --shm-size 300g --gpus all \
    --user root \
    -v /home/:/home/ \
    -v /tmp:/tmp \
    -v /data/:/data/ \
    $container /bin/bash -c "sleep infinity "


sudo docker exec -it $name bash -c "bash"