Skip to content

Commit

Permalink
PyTorch 1.6.0 update with native AMP (ultralytics#573)
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher authored Jul 31, 2020
1 parent 6cdb740 commit 1dbb6ae
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
Cython
numpy>=1.18.5
opencv-python
torch>=1.5.1
torch>=1.6
matplotlib
pillow
tensorboard
Expand All @@ -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
Expand Down

0 comments on commit 1dbb6ae

Please sign in to comment.