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