docker run -it --shm-size=16G -v $PWD/gemma_pytorch:/home/gemma -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name gemma ffa1f63239fc bash
# python -m torch.utils.collect_env