default `BASE_IMAGE` is nvcr.io/nvidia/pytorch:24.02-py3 for retro-compatibility Former-commit-id: db8d00536acb02b29d10a3d735438d194656ece3
default `BASE_IMAGE` is nvcr.io/nvidia/pytorch:24.02-py3 for retro-compatibility Former-commit-id: db8d00536acb02b29d10a3d735438d194656ece3