如果自己的电脑有多个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)
然后就是直接训练:
看结果: