torch >= 1.5

This commit is contained in:
Glenn Jocher 2020-04-22 17:54:56 -07:00
parent dda0afa22e
commit aa854ecaa9
2 changed files with 2 additions and 7 deletions

View File

@ -26,12 +26,7 @@ The https://github.com/ultralytics/yolov3 repo contains inference and training c
# Requirements # Requirements
Python 3.7 or later with all of the `pip install -U -r requirements.txt` packages including: Python 3.7 or later with all of the `pip install -U -r requirements.txt` packages including `torch >= 1.5`. Docker images come with all dependencies preinstalled. Docker requirements are:
- `torch >= 1.4`
- `opencv-python`
- `Pillow`
All dependencies are included in the associated docker images. Docker requirements are:
- Nvidia Driver >= 440.44 - Nvidia Driver >= 440.44
- Docker Engine - CE >= 19.03 - Docker Engine - CE >= 19.03

View File

@ -1,7 +1,7 @@
# pip install -U -r requirements.txt # pip install -U -r requirements.txt
numpy numpy
opencv-python >= 4.1 opencv-python >= 4.1
torch >= 1.4 torch >= 1.5
matplotlib matplotlib
pycocotools pycocotools
tqdm tqdm