Docker pyYAML>=5.3.1 fix (ultralytics#2031)

taicaile · Jan 24, 2021 · 7662310 · 7662310
1 parent e717ecb
commit 7662310
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