Description
Offer support for Nvidia-Docker. I'm not sure how feasible this is b/c it seems like docker's python package doesn't support the gpus flag (related issue w/ docker's python package). Although it does appear to be on their radar (this PR)
Use case / motivation
Support running GPU accelerated docker operators. This would also be useful to folks looking to run tasks in parallel on separate GPUs.
Thanks for opening your first issue here! Be sure to follow the issue template!
DockerOperator uses docker-py internally, so It'll not be possible to think about including It until that PR is merged.
PR merged!
@nullhack could be a good time to revisit the issue now that the PR is merged
Most helpful comment
@nullhack could be a good time to revisit the issue now that the PR is merged