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

Issues with 21.04

FROM nvcr.io/nvidia/pytorch:21.03-py3
85b6fdda · Glenn Jocher · GitHub · 4c7d686d · 85b6fdda
--- a/Dockerfile
+++ b/Dockerfile
 # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
-FROM nvcr.io/nvidia/pytorch:21.04-py3
+FROM nvcr.io/nvidia/pytorch:21.03-py3

 # Install linux packages
 RUN apt update && apt install -y zip htop screen libgl1-mesa-glx