最新文章
os.environ[‘CUDA_VISIBLE_DEVICES’]无效的解决方法
已阅: 250 场景需要指定 GPU 1 来进行训练,但 os.environ['CUDA_VISIBLE_DEVICES'] = '1' 无效原因在 torch 引入前指定 GPU 才有效解决方法创建 startup.py,在 startup.py 中先指定GPU,然后再 from tools/train import main其中,main 指的是训练主函数示例代码顺序执行,进入 startup(),然后会先设置 GPU再引入 main(),再进入 parse_args() 引入 DictAction……
(◕ܫ◕)
评论于 MMDet实例分割loss_rpn_bbox为nan但其它loss正常的解决