zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

ENV HOME=/usr/src/app (#2382) 

Set HOME environment variable per Binder requirements. 
https://github.com/binder-examples/minimal-dockerfile
This commit is contained in:
Glenn Jocher 2021-03-05 15:53:57 -08:00 committed by GitHub
parent 692e1f31dc
commit c64fe219b4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
Dockerfile
View File

@ -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 ---------------------------------------------------