@@ -1,5 +1,5 @@ | |||
# Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch | |||
FROM nvcr.io/nvidia/pytorch:20.12-py3 | |||
FROM nvcr.io/nvidia/pytorch:21.02-py3 | |||
# Install linux packages | |||
RUN apt update && apt install -y zip htop screen libgl1-mesa-glx |