| # pip install -U -r requirements.txt | |
| Cython | |
| numpy>=1.18.5 | |
| opencv-python | |
| torch>=1.5.1 | |
| matplotlib | |
| pillow | |
| tensorboard | |
| PyYAML>=5.3 | |
| torchvision>=0.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 | |
| # conda install -yc conda-forge scikit-image pycocotools tensorboard | |
| # conda install -yc spyder-ide spyder-line-profiler | |
| # conda install -yc pytorch pytorch torchvision | |
| # conda install -yc conda-forge protobuf numpy && pip install onnx==1.6.0 # https://github.com/onnx/onnx#linux-and-macos | |