From 766231050b4e7528e32aaccc98d6402a03c51ea2 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sun, 24 Jan 2021 11:54:15 -0800
Subject: [PATCH] Docker pyYAML>=5.3.1 fix (#2031)

---
 Dockerfile       | 2 +-
 requirements.txt | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index ce9cdc34f1c3..1f301b2d1e2d 100644
--- 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
index 759a100a212c..d22b42f5d786 100755
--- 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