【mmaction2 调用指定GPU】mmaction2 使用指定编号的GPU训练模型

最新推荐文章于 2024-06-02 14:59:57 发布

CSPhD-winston-杨帆

最新推荐文章于 2024-06-02 14:59:57 发布

阅读量1.6k

点赞数 2

文章标签：深度学习 python

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.csdn.net/WhiffeYF/article/details/120716036

版权

如果自己的电脑有多个GPU，如何在mmaction2中调用指定的GPU呢？

下面教你一行代码实现

tools/train.py

进入到 / mmaction2/ tools / train.py 中
找到如下代码：
在这里插入图片描述
注释掉红色框中的代码，然后在下面添加如下代码：

cfg.gpu_ids = range(1,2)

这一行的意思是，rang（1，2），那么可选的就是只有1，那么就是调用第1块GPU（注意，GPU编号从0开始）。
如果要调用第2块GPU，就改成range(2,3)
在这里插入图片描述

然后就是直接训练：
看结果：
在这里插入图片描述

CSPhD-winston-杨帆

关注

2
点赞
踩
3

收藏

觉得还不错? 一键收藏
6
评论
【mmaction2 调用指定GPU】mmaction2 使用指定编号的GPU训练模型

如果自己的电脑有多个GPU，如何在mmaction2中调用指定的GPU呢？下面教你一行代码实现tools/train.py进入到 / mmaction2/ tools / train.py 中找到如下代码：注释掉红色框中的代码，然后在下面添加如下代码：cfg.gpu_ids = range(1,2)这一行的意思是，rang（1，2），那么可选的就是只有1，那么就是调用第1块GPU（注意，GPU编号从0开始）。如果要调用第2块GPU，就改成range(2,3)然后就是直接训练：看结果
复制链接

扫一扫

评论 6

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。