Commit
•
60d5dc0
1
Parent(s):
c020875
PyTorch 1.6.0 update with native AMP (#573)
Browse files- requirements.txt +1 -4
requirements.txt
CHANGED
|
@@ -2,7 +2,7 @@
|
|
| 2 |
Cython
|
| 3 |
numpy>=1.18.5
|
| 4 |
opencv-python
|
| 5 |
-
torch>=1.
|
| 6 |
matplotlib
|
| 7 |
pillow
|
| 8 |
tensorboard
|
|
@@ -12,9 +12,6 @@ scipy
|
|
| 12 |
tqdm
|
| 13 |
# pycocotools>=2.0
|
| 14 |
|
| 15 |
-
# Nvidia Apex (optional) for mixed precision training --------------------------
|
| 16 |
-
# 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
|
| 17 |
-
|
| 18 |
# Conda commands (in place of pip) ---------------------------------------------
|
| 19 |
# conda update -yn base -c defaults conda
|
| 20 |
# conda install -yc anaconda numpy opencv matplotlib tqdm pillow ipython
|
|
|
|
| 2 |
Cython
|
| 3 |
numpy>=1.18.5
|
| 4 |
opencv-python
|
| 5 |
+
torch>=1.6
|
| 6 |
matplotlib
|
| 7 |
pillow
|
| 8 |
tensorboard
|
|
|
|
| 12 |
tqdm
|
| 13 |
# pycocotools>=2.0
|
| 14 |
|
|
|
|
|
|
|
|
|
|
| 15 |
# Conda commands (in place of pip) ---------------------------------------------
|
| 16 |
# conda update -yn base -c defaults conda
|
| 17 |
# conda install -yc anaconda numpy opencv matplotlib tqdm pillow ipython
|