# Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch | # 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 | # Install dependencies | ||||
RUN pip install --upgrade pip | RUN pip install --upgrade pip |