最近ubuntu系统跑几下就死机,怕了怕了,后面会把以前安装好的教程都发在csdn上。以免后面系统崩溃后又要重新装过而不知所措。 前段时间yolov5横空出世,然后又多出来了各种比它好的yolo算法。本次安装的是yolox,,听说是旷视研发团队搞出来的。现在我们就来看看具体的安装步骤吧。 安装系统ubuntu18.04 cuda10.1 cudnn7.6.5 安装环境通通在anaconda环境下进行
git clone https://github.com/Megvii-BaseDetection/YOLOX
conda create -n yolox python=3.8
conda activate yolox
conda install cudatoolkit==10.2.89
conda install cudnn==8.2.1.32
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch
pip install torch-1.7.1-cp38-cp38-linux_x86_64.whl
pip install torchvision-0.8.2-cp38-cp38-linux_x86_64.whl
pip install torchaudio-0.7.2-cp38-cp38-linux_x86_64.whl
cd YOLOx
pip3 install -U pip
pip3 install -r requirements.txt
python3 setup.py develop
cd apex
pip3 install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
pip3 install cython; pip3 install 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
python tools/demo.py image -n yolox-s -c weights/yolox_s.pth.tar --path assets/bus.jpg --conf 0.3 --nms 0.65 --tsize 640 --save_result
|