搜索内容

最新文章
  • os.environ[‘CUDA_VISIBLE_DEVICES’]无效的解决方法
    Python

    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……
    MWHLS 2022/08/20