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

yolov7的bug,无法指定显卡(程序默认0号卡)

**问题:**命令行参数指定的是4号卡,但实际却总是在0号卡建立进程
真抽象啊,这一步,模型被送到0号卡,但实际上,送到了4号卡(进程是在4号卡上建立的)在这里插入图片描述
解决办法:
在train.py的train()函数下,第一行选择显卡:

device=torch.device('cuda:4')

自己指定显卡即可,也可以在命令行添加一遍,不冲突,但要一样的

http://www.dtcms.com/a/2212.html

相关文章:

  • Ubuntu20.04 配置 yolov5_ros 功能包记录
  • (五)激光线扫描-位移台标定
  • Flutter项目安装到Android手机一直显示在assembledebug
  • Qt 关于mouseTracking鼠标追踪和tabletTracking平板追踪的几点官方说明
  • 关于深拷贝和浅拷贝你需要了解的内容
  • 【maven私库nexus开机自启动】
  • 【C++】面向对象编程(六)在派生类中定义一个虚函数、虚函数的静态解析
  • uniapp快速入门系列(2)- Vue基础知识
  • NPDP产品经理知识(产品创新管理)
  • Games104现代游戏引擎笔记高级ai
  • Linux 安全 - 内核提权
  • 动态代理IP常见超时原因及解决方法
  • 如何实现MySQL的增删改查
  • 【C++设计模式之迭代器模式】分析及示例
  • 微服务技术栈
  • 二叉搜索树的基础操作
  • PyTorch Lightning - LightningModule 训练逻辑 (training_step) 异常处理 try-except
  • Python—Scrapy实践项目
  • objective-c 基础学习
  • 软考 系统架构设计师系列知识点之软件架构风格(5)
  • 2023-2024年华为ICT网络赛道模拟题库
  • 【VUE3 Teleport】
  • redis分布式秒杀锁
  • 【数据结构】二叉树
  • 厌烦了iPhone默认的热点名称?如何更改iPhone上的热点名称
  • 视觉效果绝佳的制作电子宣传册的网站
  • C++交换a和b的方法
  • 导出视频里的字幕
  • ChainForge:衡量Prompt性能和模型稳健性的GUI工具包
  • 面试经典 150 题 4 —(数组 / 字符串)— 80. 删除有序数组中的重复项 II