Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

关于多卡推理 #10248

Closed
minboo opened this issue Jun 27, 2023 · 2 comments
Closed

关于多卡推理 #10248

minboo opened this issue Jun 27, 2023 · 2 comments
Assignees
Labels

Comments

@minboo
Copy link

minboo commented Jun 27, 2023

是这样的,我的服务器上有四张T4卡,但是使用paddleocr推理时只用了0号卡,其余几张卡都是空闲状态,环境变量设置了export CUDA_VISIBLE_DEVICES=0,1,2,3,请问要怎样才能利用所用GPU?麻烦大佬解答一下。

@shiyutang shiyutang added the good first issue Good for newcomers label Jun 29, 2023
@livingbody
Copy link
Contributor

推力一般用不到。不过可以试试多卡训练的方式。比如:

python3 -m paddle.distributed.launch --ips="xx.xx.xx.xx,xx.xx.xx.xx" --gpus '0,1,2,3' tools/train.py -c configs/det/det_mv3_db.yml \
     -o Global.pretrained_model=./pretrain_models/MobileNetV3_large_x0_5_pretrained

https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.6/doc/doc_ch/detection.md

@shiyutang
Copy link
Collaborator

以上回答已经充分解答了问题,如果有新的问题欢迎随时提交issue,或者在此条issue下继续回复~
我们开启了飞桨套件的ISSUE攻关活动,欢迎感兴趣的开发者参加:PaddlePaddle/PaddleOCR#10223

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

4 participants