From 6ba36265fbcf7218ff69032109c3ee74709633e1 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 22 Dec 2020 17:48:40 -0800 Subject: [PATCH] FROM nvcr.io/nvidia/pytorch:20.12-py3 (#1620) --- Dockerfile | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/Dockerfile b/Dockerfile index e514893..6727548 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,10 +1,13 @@ # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch -FROM nvcr.io/nvidia/pytorch:20.10-py3 +FROM nvcr.io/nvidia/pytorch:20.12-py3 -# Install dependencies +# Install linux packages +RUN apt update && apt install -y screen libgl1-mesa-glx + +# Install python dependencies RUN pip install --upgrade pip -# COPY requirements.txt . -# RUN pip install -r requirements.txt +COPY requirements.txt . +RUN pip install -r requirements.txt RUN pip install gsutil # Create working directory -- GitLab