weixin_46050479的博客

pytorch+cuda+onnx环境配置去年捣鼓过cuda10.2，最近捣鼓yolov5从pytorch转换为onnx时出错了，捣鼓了半天。。。这里提供一个稳定的环境配置:pytorch1.7.1+cuda11.0.03+cudnn8.0.2.39+onnx1.9+onxruntime(-gpu)1.8.0ps：前提是支持cuda11.0，用命令nvidia-smi查看设备cuda最高支持什么版本。...

pip安装onnxruntime_gpu1.6.0失败安装onnxruntime_gpu时需要对应cuda版本安装，笔者安装的cuda版本为10.2，对应onnxruntime_gpu应该安装1.5-1.6的，但是直接用命令pip install onnxruntime-gpu 安装的版本是最新版，指定版本安装pip install onnxruntime-gpu==1.6.0同样报错Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple

yolov5记录检测框中心坐标最近用到yolov5进行目标检测并确定目标在图片中的位置，直接上代码：第一步在detect.py中建立文件夹并对每张图片建立txt文件：location_center_dir = str(save_dir) + '/location_center'if not os.path.exists(location_center_dir): os.makedirs(location_center_dir)location_center_path = location_cen