Docker pyYAML>=5.3.1 fix (ultralytics#2031)

KMint1819 · Jan 24, 2021 · 9301d78 · 9301d78
1 parent ca51678
commit 9301d78
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -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
 

diff --git a/requirements.txt b/requirements.txt
@@ -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