xhost +local:
docker run --shm-size=64g --gpus all -it --rm \
-v /home/vitorguizilini/Documents/code/github/tri-ml/cosmos-predict2:/workspace \
-v /home/vguizilini/workspace/data/predict2/data:/data \
-v /home/vguizilini/workspace/data/predict2/datasets:/workspace/datasets \
-v /home/vguizilini/workspace/data/predict2/checkpoints:/workspace/checkpoints \
-v /dev/null:/dev/raw1394 \
-e DISPLAY=${DISPLAY} \
-e XAUTHORITY \
-e NVIDIA_DRIVER_CAPABILITIES=all \
--privileged \
--ipc=host \
--network=host \
cosmos-predict2-local:latest