Unverified 提交 60d5dc02 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

PyTorch 1.6.0 update with native AMP (#573)

上级 c020875b
...@@ -2,7 +2,7 @@ ...@@ -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 ...@@ -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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论