device=cuda:0 means running on CPU only?

Question

device=cuda:0 means running on CPU only?

Closed this issue 4 years ago · 1 comments

Hi,
I'm trying to replica this repository and wonder if current setting is for CPU only since I search for "cuda" in repository and see most parameter as "0". Thank you if you can help to comment.

Answer 1 · 2021-01-14T10:11:54.000Z

It means that you use only the first GPU if more than one are found.

…

On Thu, Jan 14, 2021, 11:15 michaelcwang2 ***@***.***> wrote: Hi, I'm trying to replica this repository and wonder if current setting is for CPU only since I search for "cuda" in repository and see most parameter as "0". Thank you if you can help to comment. — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub <#7>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ADFQGYS4YL36BWJG56XDDJDSZ2Y2BANCNFSM4WCDX4DQ> .