FROM nvcr.io/nvidia/pytorch:20.06-py3

Dockerfile

 # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
-FROM nvcr.io/nvidia/pytorch:20.03-py3
+FROM nvcr.io/nvidia/pytorch:20.06-py3
 RUN pip install -U gsutil
 
 # Create working directory