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

Posted 2021-10-31 计算机视觉-杨帆

如果自己的电脑有多个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)

然后就是直接训练：
看结果：