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

在多块昆仑R200的服务器跑代码时,怎么指定使用某块R200? #9367

Open
xc005 opened this issue Nov 5, 2024 · 1 comment
Open
Assignees
Labels
others unknown issue type

Comments

@xc005
Copy link

xc005 commented Nov 5, 2024

问题描述

看R200相关文档,只看到多卡训练指定R200卡号,但单卡推理时没有,默认是用的0号,但若不使用0号卡,怎么指定呀?

有没有类似英伟达卡的"CUDA_VISIBLE_DEVICES=1 python xxx.py"的指定方法?

@xc005 xc005 added the others unknown issue type label Nov 5, 2024
@DrownFish19
Copy link
Collaborator

可以使用paddle.distributed.launch启动,通过--devices指定运行设备。

python -u -m paddle.distributed.launch --devices run_xxx.py

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

No branches or pull requests

3 participants