torch >= 1.5
This commit is contained in:
parent
dda0afa22e
commit
aa854ecaa9
|
@ -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
|
||||||
|
|
||||||
|
|
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue