yolox
安装
安装 YOLOX
git clone git@github.com:Megvii-BaseDetection/YOLOX.git
cd YOLOX
python3 -m pip install -U pip && python3 -m pip install -r requirements.txt
python3 -m pip install -v -e . # or python3 setup.py develop
安装 apex
如果你不想训练模型,请跳过此步骤。
git clone https://github.com/NVIDIA/apex
cd apex
python3 -m pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
使用 最新
gcc 10.3.0
编译会出错 ,建议使用低于10
的版本当前使用的
apex
版本为 0c2c6ee安装
apex
前需要安装cuda
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600 sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /" | sudo tee -a /etc/apt/sources.list.d/cuda.list sudo apt update sudo apt -y install cuda-11-1
当前支持
cuda
最高版本为11.1
安装 pycocotools
python3 -m pip install cython
python3 -m pip install 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'