-
I would like to know how to specify which device is used when running this application. Is there a line of code I can add to the start batch file that lets me specify if I would like to use CPU, GPU.0, GPU.1, or GPU.*? Right now it seems to be defaulting to "cpu". |
Beta Was this translation helpful? Give feedback.
Answered by
rupeshs
Jan 10, 2024
Replies: 1 comment 1 reply
-
@GuiltyImp For that you need to set the DEVICE environment variable. |
Beta Was this translation helpful? Give feedback.
1 reply
Answer selected by
GuiltyImp
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
@GuiltyImp For that you need to set the DEVICE environment variable.
If you are on Windows
set DEVICE=GPU.0
For Linux
export DEVICE=GPU.0