yolox 安装环境 ubuntu

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

  ubuntu 18.04 安装 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'