测试单张图片时报错: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU. about awesome-backbones

[Fafa-DL]

其实是有这个逻辑的，可以看截图代码29-32行，不过现在看来没生效，因为args.device无论在终端是否指定（未指定默认为cuda）都为True，感谢反馈

[whitewatercn]

将tools/single_test.py中的

    parser.add_argument(
        '--device', default='cuda', help='Device used for inference')

手动改为

    parser.add_argument(
        # '--device', default='cuda', help='Device used for inference')
        '--device', default='cpu', help='Device used for inference')

即可

期待up加一个自动识别功能，我看不太懂，只能做到这里了