# Run the container with GPU support and mount necessary directories
# docker run --shm-size=64g --gpus all -it --rm \
# -v /home/vguizilini/workspace/cosmos-predict2:/workspace \
# -v /home/vguizilini/workspace/data/predict2/datasets:/workspace/datasets \
# -v /home/vguizilini/workspace/data/predict2/checkpoints:/workspace/checkpoints \
# cosmos-predict2-local:latest

# docker run --shm-size=64g --gpus all -it --rm \
# -v $HOME/workspace/cosmos-predict2:/workspace \
# -v $HOME/workspace/data/predict2/data:/data \
# -v /home/junjie.ye/workspace/data/predict2/datasets:/data_junjie \
# -v $HOME/workspace/data:/data_all \
# -v $HOME/workspace/data/predict2/datasets:/workspace/datasets \
# -v $HOME/workspace/data/predict2/checkpoints:/workspace/checkpoints \
# cosmos-predict2-local:latest

docker run --shm-size=64g --gpus all -it --rm \
-v $HOME/workspace/cosmos-predict2:/workspace \
-v $HOME/workspace/data/predict2/data:/data \
-v /home/junjie.ye/workspace/data/predict2/datasets:/data_junjie \
-v $HOME/workspace/data:/data_all \
-v $HOME/workspace/data/predict2/datasets:/workspace/datasets \
-v $HOME/workspace/data/predict2/checkpoints:/workspace/checkpoints \
cosmos-predict2-local:latest

# Verify setup inside container
# python /workspace/scripts/test_environment.py