|
# pip install -U -r requirements.txt |
|
numpy==1.17 |
|
opencv-python |
|
torch >= 1.5 |
|
matplotlib |
|
pycocotools |
|
tqdm |
|
pillow |
|
tensorboard |
|
pyyaml |
|
|
|
|
|
# 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 # https://github.com/onnx/onnx#linux-and-macos |
|
|