yolov5 / requirements.txt
glenn-jocher's picture
PyTorch 1.6.0 update with native AMP (#573)
60d5dc0 unverified
raw
history blame
660 Bytes
# pip install -U -r requirements.txt
Cython
numpy>=1.18.5
opencv-python
torch>=1.6
matplotlib
pillow
tensorboard
PyYAML>=5.3
torchvision>=0.6
scipy
tqdm
# pycocotools>=2.0
# Conda commands (in place of pip) ---------------------------------------------
# conda update -yn base -c defaults conda
# conda install -yc anaconda numpy opencv matplotlib tqdm pillow ipython
# conda install -yc conda-forge scikit-image pycocotools tensorboard
# conda install -yc spyder-ide spyder-line-profiler
# conda install -yc pytorch pytorch torchvision
# conda install -yc conda-forge protobuf numpy && pip install onnx==1.6.0 # https://github.com/onnx/onnx#linux-and-macos