From 895a13dd13000ecffe31b7ecfbf8dfc828ccf342 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 5 Mar 2021 15:53:57 -0800
Subject: [PATCH] ENV HOME=/usr/src/app (#2382)

Set HOME environment variable per Binder requirements.
https://github.com/binder-examples/minimal-dockerfile
---
 Dockerfile | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 1a8fe2e72885..e1b40c2d15c6 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -5,8 +5,8 @@ FROM nvcr.io/nvidia/pytorch:21.02-py3
 RUN apt update && apt install -y zip htop screen libgl1-mesa-glx
 
 # Install python dependencies
-RUN python -m pip install --upgrade pip
 COPY requirements.txt .
+RUN python -m pip install --upgrade pip
 RUN pip install --no-cache -r requirements.txt gsutil notebook
 
 # Create working directory
@@ -16,11 +16,8 @@ WORKDIR /usr/src/app
 # Copy contents
 COPY . /usr/src/app
 
-# Copy weights
-#RUN python3 -c "from models import *; \
-#attempt_download('weights/yolov5s.pt'); \
-#attempt_download('weights/yolov5m.pt'); \
-#attempt_download('weights/yolov5l.pt')"
+# Set environment variables
+ENV HOME=/usr/src/app
 
 
 # ---------------------------------------------------  Extras Below  ---------------------------------------------------