mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Fix pip requirements issues

Browse Source
This commit is contained in:
Adam 2020-06-14 18:18:31 +10:00
parent 099e6f5ebd
commit faa6aeaa20
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
requirements.txt
View File

@ -1,4 +1,5 @@
# pip install -U -r requirements.txt # pip install -U -r requirements.txt
Cython
numpy==1.17 numpy==1.17
opencv-python opencv-python
torch >= 1.5 torch >= 1.5
@ -8,7 +9,8 @@ tqdm
pillow pillow
tensorboard tensorboard
pyyaml pyyaml
torchvision
scipy
# Nvidia Apex (optional) for mixed precision training -------------------------- # Nvidia Apex (optional) for mixed precision training --------------------------
# git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" . --user && cd .. && rm -rf apex # git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" . --user && cd .. && rm -rf apex