paddleocr 这个库如何在代码里面设定GPU的使用卡号?
收藏
我的代码情况是一个多进程下使用paddleocr,我的服务器是有多张GPU卡,
然后我想实现的效果类似:我第一条进程使用GPU的0号卡,第二条进程使用GPU的1号卡,第三条进程使用GPU的2号卡。
请教一下大神我能用什么办法去实现这样的效果?
我自己尝试过将os.environ['CUDA_VISIBLE_DEVICES'] = '0' 这放到代码的主函数里面,但这样这命令是不起效的,这个只能放在导入库的时候设置才行,所以无法实现我想要的指定进程去调用不同的GPU的需求。
如果有实现过的大神能给我留一下言,教教我吗?
0
收藏
请登录后评论