PyTorch 1.6.0 update with native AMP (#573)
This commit is contained in:
1 parent
c020875b17
commit
60d5dc0294
1 file changed
+1
-4
+1
-4
@@ -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
|
||||
|
||||
Reference in new issue
Block a user