EasyOCR 使用多个GPU

4xrmg8kj  于 6个月前  发布在  其他
关注(0)|答案(2)|浏览(51)

大家好
我正在运行easyocr多GPU,但它只使用了一个GPU。有什么建议如何使用多个GPU进行推理?

bt1cpqcv

bt1cpqcv2#

是否有解决方法?我不断收到这个消息:模块必须在设备cuda:0(device_ids[0])上具有其参数和缓冲区,但在设备cuda:2上找到了其中一个。当尝试在与默认GPU不同的GPU中使用它时,似乎找不到一种实现多个GPU的方法。

相关问题