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.

Sign up for GitHub

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

Jump to bottom

如何指定使用GPU=[3,4,5,6]训练，不使用GPU=0？ #680

Open

renliao opened this issue May 18, 2024 · 1 comment

renliao commented May 18, 2024

亲爱的作者，是配置--local_rank=3吗？我配置了以后发现还是会使用GPU=0.

Windrain7 commented Nov 3, 2024

最简单的方法，在运行命令前加上CUDA_VISIBLE_DEVICES=[3,4,5,6]

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment