docker run -it --shm-size=100G -v $PWD/RepViT:/home/RepViT -v /parastor/DL_DATA/HOT:/home/HOT --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name repvit f6b99c8a0f01 bash # python -m torch.utils.collect_env
