Updated requirements for python 3.8. Modified tensorflow nn module to…

… adjust for python 3.8. See: ultralytics/yolov5#6948
theoway · Jun 16, 2022 · 97bd2cb · 97bd2cb
1 parent eb30044
commit 97bd2cb
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -14,12 +14,12 @@ flake8
 yapf
 isort
 imageio
-tensorflow==1.15
-tensorflow-gpu==1.15
+tensorflow>=1.15
+tensorflow-gpu>=1.15
 scikit-learn==0.22.2
-keras==2.3.1
+keras>=2.3.1
 pycocotools
 seaborn
-torch==1.10.1+cu113
-torchaudio==0.10.1+cu113
-torchvision==0.11.2+cu113
+torch==1.11.0+cu113
+torchaudio==0.11.0+cu113
+torchvision==0.12.0s+cu113