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

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

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

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

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

相关文章:

  • 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)
  • 重庆荣昌区委区政府再设“答谢宴”,邀请800余名志愿者机关食堂用餐
  • 夜读丨喜马拉雅山的背夫
  • 经彩申城!上海网络大V沙龙活动走进闵行
  • 菲护卫艇企图侵闯中国黄岩岛领海,南部战区:依法依规跟踪监视、警告驱离
  • 外交部:解放军参加红场阅兵体现了中方对历史的尊重和铭记
  • 外交部介绍中国赞比亚共同举办人工智能能力建设主题活动情况