原文链接： blog.csdn.net/WhiffeYF/ar…
如果自己的电脑有多个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)
然后就是直接训练：
看结果：
