diff --git a/requirements.txt b/requirements.txt index a6d8b92e..dfb5cd90 100755 --- a/requirements.txt +++ b/requirements.txt @@ -7,11 +7,14 @@ pycocotools tqdm Pillow -# Tensorboard pip requirements ------------------------------------------------- +# 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 + +# Tensorboard (optional) pip requirements -------------------------------------- # tb-nightly # future -# Equivalent conda commands ---------------------------------------------------- +# Conda commands (in place of pip) --------------------------------------------- # conda update -n base -c defaults conda # conda install -yc anaconda numpy opencv matplotlib tqdm pillow ipython future # conda install -yc conda-forge scikit-image pycocotools onnx tensorboard