使用PyCharm及其等价命令行运行YoloV5 (启用GPU加速)

目录

前提

1.copy过来后要删除其根目录下的.idea folder：

2.删除下面folder里面的所有内容：

3.在Pycharm中重新建立一个Virtual enviorenment:

4.在Pycharm的终端安装YoloV5 的requirements：

5.可能会遇到下面下面问题，解决方案见下：

6.如果你的机器有GPU显卡，则可按如下启用GPU加速：

另外，可用pyCharm的等价命令行启动Yolo V5：

---

前提

系统环境：Windows10.

首先安装Windows 64bit 的Python 3.6.2。

再安装PyCharm的社区版.

然后从官网Clone Yolo V5.

我是从从别处copy的整个Yolo v5 整个folder，如果不是copy的source code ，从下面第3步开始:

1.copy过来后要删除其根目录下的.idea folder：

因为这个folder会留有以前的环境信息，导致run不起来：

    C:\XXX\yolov5\.idea\workspace.xml （

        行 54:       <option name="SDK_HOME" value="C:\XXX\yolov5-master\venv\Scripts\python.exe" />

2.删除下面folder里面的所有内容：

    C:\XXX\yolov5\venv

如果不是copy的source code ，不用做以上两步。

3.在Pycharm中重新建立一个Virtual enviorenment:

4.在Pycharm的终端安装YoloV5 的requirements：

5.可能会遇到下面下面问题，解决方案见下：

 building 'Levenshtein._levenshtein' extension

  error: Microsoft Visual C++ 14.0 or greater is required. Get it with "Microsoft C++ Build Tools": Microsoft C++ Build Tools - Visual Studio

请参考下面连接解决，实际是需要安装C++ build工具：

error: Microsoft Visual C++ 14.0 or greater is required. Get it with "Microsoft C++ Build Tools": https://visualstudio.microsoft.com/visual-cpp-build-tools/ - Microsoft Q&A

6.如果你的机器有GPU显卡，则可按如下启用GPU加速：

假如你有GPU显卡，先跑一下yolov 5的测试，如果出现如下错误，则说明可能pytorch未安装正确：

            AssertionError: CUDA unavailable, invalid device 0 requested

解决如下，在Pycharm的终端安装：

pip3 install torch==1.10.0+cu102 torchvision==0.11.1+cu102 torchaudio===0.10.0+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

上面命令行来自 <Start Locally | PyTorch> ，如下：

如上步骤，则可以使用Pycharm run Yolo V5的detect.py，记得设置该文件参数加上--device 0，表示启用GPU. 

另外，可用pyCharm的等价命令行启动Yolo V5：

先在键盘上按Win+R 启动终端cmd：

直接进入yolov5下面的folder，以激活PyCharm的虚拟环境 :

C:\XXX\yolov5\venv\Scripts>activate.bat

(venv) C:\XXX\yolov5\venv\Scripts>cd ..

假如你用了Mark Directory as Sources Root:

那么请使用以下，达到等价效果：

(venv) C:\XXX\yolov5>set PYTHONPATH=%PYTHONPATH%;C:\XXX\yolov5\PaddleOCR

检测一下上面的设置：

(venv) C:\XXX\yolov5>echo %PYTHONPATH%

%PYTHONPATH%;C:\XXX\yolov5\PaddleOCR

最后命令行启动Yolo V5的测试：

(venv) C:\XXX\yolov5>python detect.py --weights yolov5s.pt --source 0 --device 0