mirror of
https://github.com/microsoft/autogen.git
synced 2025-08-31 12:00:11 +00:00

The DockerCommandLineCodeExecutor doesn't currently offer GPU support. By simply using DeviceRequest from the docker python API, these changes expose GPUs to the docker container and provide the ability to execute CUDA-accelerated code within autogen. ## Related issue number Closes: #6302 ## Checks - [x] I've included any doc changes needed for <https://microsoft.github.io/autogen/>. See <https://github.com/microsoft/autogen/blob/main/CONTRIBUTING.md> to build and test documentation locally. - [x] I've added tests (if relevant) corresponding to the changes introduced in this PR. - [x] I've made sure all auto checks have passed. --------- Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com>