Skip to content

Force CPU #1266

Answered by hbredin
doublex asked this question in Q&A
Feb 21, 2023 · 2 comments
Discussion options

You must be logged in to vote

Short answer: not yet.
Long answer: in current develop branch, pipelines run on CPU by default and can be sent to a particular device using pipeline.to(torch.device("cuda")).

Replies: 2 comments

Comment options

You must be logged in to vote
0 replies
Answer selected by doublex
Comment options

You must be logged in to vote
0 replies
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants