Yolo integration #86
Answered
by
omichel
Siddharth220903
asked this question in
Q&A
Yolo integration
#86
-
|
Hi! Could you please specify is it possible to integrate yolo network with the gpu that would be used to run the controller during evaluation? Provided we have pre-trained weights for yolo. |
Beta Was this translation helpful? Give feedback.
Answered by
omichel
May 22, 2023
Replies: 1 comment
-
|
You can integrate whatever library you need in your docker image. You should simply add the "compute" flag so that it will allow the use of the CPU for compute. Normally adding something like: ENV NVIDIA_DRIVER_CAPABILITIES graphics,compute,utilityin your Dockerfile, should enable GPU compute inside your docker. |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
omichel
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
You can integrate whatever library you need in your docker image. You should simply add the "compute" flag so that it will allow the use of the CPU for compute. Normally adding something like:
ENV NVIDIA_DRIVER_CAPABILITIES graphics,compute,utilityin your Dockerfile, should enable GPU compute inside your docker.