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
Python 3.7 or later with all of the `pip install -U -r requirements.txt` packages including:
- `torch >= 1.4`
- `opencv-python`
- `Pillow`
All dependencies are included in the associated docker images. Docker requirements are:
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:
- Nvidia Driver >= 440.44
- Docker Engine - CE >= 19.03

View File

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