首页 PaddleOCR 帖子详情
paddleocr 这个库如何在代码里面设定GPU的使用卡号?
收藏
快速回复
PaddleOCR 问答OCR 513 0
paddleocr 这个库如何在代码里面设定GPU的使用卡号?
收藏
快速回复
PaddleOCR 问答OCR 513 0

我的代码情况是一个多进程下使用paddleocr,我的服务器是有多张GPU卡,

然后我想实现的效果类似:我第一条进程使用GPU的0号卡,第二条进程使用GPU的1号卡,第三条进程使用GPU的2号卡。

请教一下大神我能用什么办法去实现这样的效果?

我自己尝试过将os.environ['CUDA_VISIBLE_DEVICES'] = '0' 这放到代码的主函数里面,但这样这命令是不起效的,这个只能放在导入库的时候设置才行,所以无法实现我想要的指定进程去调用不同的GPU的需求。

如果有实现过的大神能给我留一下言,教教我吗?

0
收藏
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户