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

How can I run this on CPU? #23

Open
cywuuuu opened this issue Nov 15, 2024 · 1 comment
Open

How can I run this on CPU? #23

cywuuuu opened this issue Nov 15, 2024 · 1 comment

Comments

@cywuuuu
Copy link

cywuuuu commented Nov 15, 2024

I tried these following steps https://recbole.io/docs/v0.1.2/user_guide/config_settings.html

and setting

...
use_gpu: False
device: -1

in yamls
But still I find it running on my gpu? I need some suggestions, as i am debugging a CUDA device-end error.

@hyp1231
Copy link
Member

hyp1231 commented Jan 5, 2025

Could you please try:

python finetune.py -d Scientific --gpu_id=~

In this way, gpu_id is None and RecBole could set the device as CPU following:

https://github.com/RUCAIBox/RecBole/blob/68ef746b34c5cee28de0ddbdf6aaa29becfa314b/recbole/config/configurator.py#L490-L494

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

No branches or pull requests

2 participants