diff --git a/README.md b/README.md index 17634800..19402d4f 100755 --- a/README.md +++ b/README.md @@ -14,7 +14,7 @@ The https://github.com/ultralytics/yolov3 repo contains inference and training c Python 3.7 or later with the following `pip3 install -U -r requirements.txt` packages: - `numpy` -- `torch` +- `torch >= 1.0.0` - `opencv-python` # Training diff --git a/requirements.txt b/requirements.txt index 2d57893b..eade9a5b 100755 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ # pip3 install -U -r requirements.txt numpy opencv-python -torch +torch >= 1.0.0 matplotlib