Hello Guys. I am running into an issue and the doc...
# ask-community
a
Hello Guys. I am running into an issue and the docs don't provide a clear answer. I am starting a worker as follows:
prefect worker start -t docker --pool <pool-name>
I want this worker to utilize the underlying GPU so that when the docker image is ran in this worker, it has access to the cuda files that comes with the gpu. How can I accomplish this?
w
This is not strictly speaking anything to do with Prefect, you will have more success looking for a solution looking up generic docker + cuda tutorials. This should get you started: https://medium.com/@albertqueralto/enabling-cuda-capabilities-in-docker-containers-51a3566ad014
Sorry for misunderstanding. If you can create a docker image that works with cuda, and you can run a docker flow in a docker container, what is the missing bit for you to put this together?
a
so on an ec2 its an amazon linux machine, i guess it would be different if I am running my docker worker on a windows machine
your answer was actually helpful
👍 1