瀏覽代碼

PyTorch 1.6.0 update with native AMP (#573)

5.0
Glenn Jocher GitHub 4 年之前
父節點
當前提交
60d5dc0294
沒有發現已知的金鑰在資料庫的簽署中 GPG 金鑰 ID: 4AEE18F83AFDEB23
共有 1 個檔案被更改,包括 1 行新增4 行删除
  1. +1
    -4
      requirements.txt

+ 1
- 4
requirements.txt 查看文件

@@ -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

Loading…
取消
儲存