From 1dbb6ae1eaa2a608d483bb8e684374f569c5f5b3 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 31 Jul 2020 10:54:08 -0700
Subject: [PATCH] PyTorch 1.6.0 update with native AMP (#573)

---
 requirements.txt | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index c3926610d4e6..18a5461e71ae 100755
--- a/requirements.txt
+++ b/requirements.txt
@@ -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