diff --git a/requirements.txt b/requirements.txt
index 2b447aef4d39..759a100a212c 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>=3.13
+PyYAML==5.4.1
 scipy>=1.4.1
 tensorboard>=2.2
 torch>=1.7.0