Skip to content

Deploy model on multiple GPU #141

Answered by lkevinzc
pl8787 asked this question in Q&A
Discussion options

You must be logged in to vote

Hi @pl8787 ! You could pass CUDA_VISIBLE_DEVICES environment variable to the worker processes, so that each worker will occupy the specified GPU device and run in parallel to serve the requests.

See https://mosecorg.github.io/mosec/example/env/ for an example.

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by kemingy
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