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

Dockerfile

 # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
-FROM nvcr.io/nvidia/pytorch:20.08-py3
+FROM nvcr.io/nvidia/pytorch:20.09-py3
 
 # Install dependencies
 RUN pip install --upgrade pip