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

Docker pyYAML>=5.3.1 fix (#2031)

This commit is contained in:
Glenn Jocher 2021-01-24 11:54:15 -08:00 committed by GitHub
parent c76c607265
commit 3551b072b3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -5,7 +5,7 @@ FROM nvcr.io/nvidia/pytorch:20.12-py3
RUN apt update && apt install -y screen libgl1-mesa-glx
# Install python dependencies
RUN pip install --upgrade pip
RUN python -m pip install --upgrade pip
COPY requirements.txt .
RUN pip install -r requirements.txt gsutil

2
requirements.txt
View File

@ -6,7 +6,7 @@ matplotlib>=3.2.2
numpy>=1.18.5
opencv-python>=4.1.2
Pillow
PyYAML==5.4.1
PyYAML>=5.3.1
scipy>=1.4.1
tensorboard>=2.2
torch>=1.7.0