Update requirements.txt

pull/56/head
Glenn Jocher 2020-06-14 10:23:33 -07:00 committed by GitHub
parent f47ca2085c
commit 1374581929
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions

View File

@ -1,16 +1,16 @@
# pip install -U -r requirements.txt # pip install -U -r requirements.txt
Cython Cython
numpy==1.17 numpy
opencv-python opencv-python
torch >= 1.5 torch >= 1.5
matplotlib matplotlib
pycocotools
tqdm
pillow pillow
tensorboard tensorboard
pyyaml pyyaml
torchvision torchvision
scipy scipy
tqdm
'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
# Nvidia Apex (optional) for mixed precision training -------------------------- # 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 # 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