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

PyTorch 1.6.0 update with native AMP (#573)

Browse Source
This commit is contained in:
Glenn Jocher 2020-07-31 10:54:08 -07:00 committed by GitHub
parent c020875b17
commit 60d5dc0294
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
requirements.txt
View File

@ -2,7 +2,7 @@
Cython
numpy>=1.18.5
opencv-python
torch>=1.5.1
torch>=1.6
matplotlib
pillow
tensorboard
@ -12,9 +12,6 @@ scipy
tqdm
# pycocotools>=2.0
# 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
# Conda commands (in place of pip) ---------------------------------------------
# conda update -yn base -c defaults conda
# conda install -yc anaconda numpy opencv matplotlib tqdm pillow ipython