一,环境的搭建
环境
win10
python 3.11
cmake
pytorch
pycharm
过程
首先安装好一个pycharm,这里就不一一叙述了。
其次,选择好一个python版本,是关键所在。有些YOLO的版本,并不支持很高的python版本,博主选用的是python3.11版本。经过实际的测试,这个版本比较合适。
pythonpycharmPyCharmpythonpython_14">注意,配置到这里,python环境和pycharm都已经安装,这里从PyCharm新建一个项目,python建议配置一个虚拟的环境,这样不会影响系统级别的python依赖。
(一)首先就是安装cmake的环境。直接从cmake下载,然后path配置cmake路径即可。
(二)完成cmake安装后,进行onnxsim安装,这个库,会涉及到yolo转换onnx,同时也是依赖cmake才能正常安装。
相关指令:pip install onnxsim
(三)安装pytorch版本
pip install torch2.0.1+cu117 torchvision0.15.2+cu117 torchaudio==