docker / roadmap

Welcome to the Public Roadmap for All Things Docker! We welcome your ideas.
https://github.com/orgs/docker/projects/51
Creative Commons Zero v1.0 Universal
1.74k stars 261 forks source link

Docker Desktop in Linux should support GPU and/or nvidia runtime #497

Open Davidnet opened 1 year ago

Davidnet commented 1 year ago

Tell us about your request I love docker desktop but many workflows in the ML/AI space requires the use of a hardware accelerator, e.g GPU, currently due to the virtualization in Linux, docker desktop does not support GPU accelerated containers but it is supported in WSL2.

Which service(s) is this request for? Docker Desktop

Tell us about the problem you're trying to solve. What are you trying to do, and why is it hard? Running a neural network using a ML framework in a hardware accelerated device

Are you currently working around the issue? Using docker engine and/or moby

Additional context Most common installation: https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html

guidodinello commented 2 days ago

Hello! I hope you all don't mind me checking in on this issue. I was wondering if there are any updates on this, is it currently in development or part of a future roadmap? Thank you for your time!