当前位置: 首页 > news >正文

yolov7_Obb环境安装

下载obb代码之后,除了安装python和pytorch环境,由于还需要编译nms部分的c++代码,因此还需要安装Visual Studio.

这里推荐安装Visual Studio2019版本。

然后在系统环境中配置环境变量

C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.29.30133\bin\Hostx64\x64

C:\Program Files (x86)\Windows Kits\10\bin\10.0.19041.0\x64

C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.29.30133\include;

C:\Program Files (x86)\Windows Kits\10\Include\10.0.19041.0\ucrt;

C:\Program Files (x86)\Windows Kits\10\Include\10.0.19041.0\um;

C:\Program Files (x86)\Windows Kits\10\Include\10.0.19041.0\cppwinrt;

C:\Program Files (x86)\Windows Kits\10\Include\10.0.19041.0\shared;

C:\Program Files (x86)\Windows Kits\10\Include\10.0.19041.0\winrt

C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.29.30133\lib\x64;

C:\Program Files (x86)\Windows Kits\10\Lib\10.0.19041.0\ucrt\x64;

C:\Program Files (x86)\Windows Kits\10\Lib\10.0.19041.0\um\x64;

C:\Program Files (x86)\Windows Kits\10\Lib\10.0.19041.0\ucrt_enclave\x64

然后cmd打开命令窗口,输入cl.exe,查看输出以下内容为Visual Studio2019安装成功。

 

可以写一段代码测试是否可以编译成功。

#include <iostream>

int main()

{

  std::cout << "Hello World" << std::endl;

  return 0;

}

// 官方用例

将以上内容保存在一个名称为main.cpp文本里面。然后当前命令框输入cl main.cpp进行编译。

出现以下报错fatal error C1034: iostream: 不包括路径集

 

这个一般是因为没有在c++的环境里面去运行,我们可以找到安装环境中的VsDevCmd.bat,找不到路径的可以使用Everything对路径进行查找。

然后将该文件直接拖到命令行里面,可以看到命令行环境就变化了。我这里是安装在D盘下。

直接拖进来

回车

出现以下内容就ok了,再输入刚刚的内容。

 

完成生成。

接下来是安装python pytorch和cuda的环境,这三者要相互对应,pytorch对应的如果安装的cuda11.1或cuda11.8就无法匹配相对于的nvidia的cuda了,然后我这里安装了cuda11.3,nvidia的cuda也是11.3

Nvidia的cuda下载地址:

CUDA Toolkit 12.3 Update 2 Downloads | NVIDIA Developer

选择历史版本:

下载好双击安装即可,这里选择自定义:

然后只选择cuda即可。

完成安装之后,我们cmd弹出的命令行输入nvcc -V查看是否安装好了cuda

这说明已经安装好了,版本cuda11.3

然后还要安装cudnn

cuDNN Archive | NVIDIA Developer

选择合适的cudnn

然后将里面的内容都复制到当前目录下,替换即可。

然后我们验证cudnn是否成功

复制完后,在当前目录下进入 extras -> demo_suite,可以看到有 bandwidthTest.exe 和 deviceQuery.exe

出现以上内容说明成功了。

也可以安装多个cuda只要注意在系统环境变量中开头的顺序即可。

然后pytorch的话我就安装了1.10.1版本的

地址:https://pytorch.org/get-started/previous-versions/

最后我们在以下代码界面打开cmd进行编译

命令窗口内输入

python setup.py build_ext --inplace

如果出现报错:ImportError: cannot import name 'obb_nms' from 'utils.nms_rotated'

一般是环境没安装正确,cuda版本对应不上。

然后解决之后,再运行,报

nvcc fatal : Unsupported gpu architecture 'compute_86'

说明计算机算力和cuda不匹配,我们可以在环境变量中设置算法。

比如

在系统环境变量中添加TORCH_CUDA_ARCH_LIST

我这里是30系列显卡,所以我值设置8.0

最后再运行

python setup.py build_ext --inplace

运行不报错即可。

最后我们训练模型也可以正常训练了。

 

相关文章:

  • YOLOv8改进 | 检测头篇 | 利用DBB重参数化模块魔改检测头实现暴力涨点 (支持检测、分割、关键点检测)
  • 红黑树(RBTree)
  • 如何在 Ubuntu 中更改时区设置
  • 谷歌地图多个maker标记点击显示当前信息弹窗infowindow
  • ardupilot开发 --- 固件定制(OEM) 篇
  • 任务12:使用Hadoop Streaming解压NCDC天气原始数据
  • Resize:最近邻插值、双线性插值、双三次插值
  • 1.环境部署
  • export 是一个在 Unix 和类 Unix 系统(比如 Linux 和 macOS)中常用的 shell 命令,主要用于设置或导出环境变量。
  • C++——STL标准模板库——容器详解——set
  • 亚马逊云科技 WAF 部署小指南(五):在客户端集成 Amazon WAF SDK 抵御 DDoS 攻击...
  • Abp vNext(一)说明
  • 鸿蒙系列--数据管理
  • 百亿大模型在GTX1060上的高效运行优化
  • 2024 CKA 题库 | 10、创建 PV
  • 《网管员必读——网络组建》(第2版)电子课件下载
  • 【翻译】babel对TC39装饰器草案的实现
  • canvas 绘制双线技巧
  • Java 23种设计模式 之单例模式 7种实现方式
  • React-redux的原理以及使用
  • scrapy学习之路4(itemloder的使用)
  • SQLServer插入数据
  • TypeScript实现数据结构(一)栈,队列,链表
  • 服务器从安装到部署全过程(二)
  • 给Prometheus造假数据的方法
  • 记一次和乔布斯合作最难忘的经历
  • 时间复杂度与空间复杂度分析
  • 微信公众号开发小记——5.python微信红包
  • 协程
  • 在Docker Swarm上部署Apache Storm:第1部分
  • Java数据解析之JSON
  • 带你开发类似Pokemon Go的AR游戏
  • ​3ds Max插件CG MAGIC图形板块为您提升线条效率!
  • ​LeetCode解法汇总2182. 构造限制重复的字符串
  • #162 (Div. 2)
  • $.ajax,axios,fetch三种ajax请求的区别
  • (14)目标检测_SSD训练代码基于pytorch搭建代码
  • (4.10~4.16)
  • (C语言)球球大作战
  • (附源码)计算机毕业设计SSM疫情居家隔离服务系统
  • (学习日记)2024.03.25:UCOSIII第二十二节:系统启动流程详解
  • *Algs4-1.5.25随机网格的倍率测试-(未读懂题)
  • .form文件_SSM框架文件上传篇
  • .NET Core 实现 Redis 批量查询指定格式的Key
  • .netcore如何运行环境安装到Linux服务器
  • .NET性能优化(文摘)
  • /dev/sda2 is mounted; will not make a filesystem here!
  • ??在JSP中,java和JavaScript如何交互?
  • @vue/cli脚手架
  • [ 蓝桥杯Web真题 ]-布局切换
  • [ 手记 ] 关于tomcat开机启动设置问题
  • [ 英语 ] 马斯克抱水槽“入主”推特总部中那句 Let that sink in 到底是什么梗?
  • [Android] Implementation vs API dependency
  • [C# WPF] DataGrid选中行或选中单元格的背景和字体颜色修改
  • [Delphi]一个功能完备的国密SM4类(TSM4)[20230329更新]