diff --git a/Dockerfile b/Dockerfile index da91e035..f21cf49e 100644 --- a/Dockerfile +++ b/Dockerfile @@ -2,7 +2,7 @@ FROM nvcr.io/nvidia/pytorch:20.03-py3 # Install dependencies (pip or conda) -RUN pip install -U gsutil thop +RUN pip install -U gsutil # RUN pip install -U -r requirements.txt # RUN conda update -n base -c defaults conda # RUN conda install -y -c anaconda future numpy opencv matplotlib tqdm pillow