Please use the image nvcr.io/nvidia/pytorch:21.11-py3 when you want to launch it