Skip to content
This repository was archived by the owner on Sep 5, 2023. It is now read-only.

Accelerator support (GPU) #68

Closed
stefdoerr opened this issue Nov 9, 2022 · 2 comments · Fixed by #72
Closed

Accelerator support (GPU) #68

stefdoerr opened this issue Nov 9, 2022 · 2 comments · Fixed by #72
Labels
api: batch Issues related to the googleapis/python-batch API. external This issue is blocked on a bug with the actual product. type: docs Improvement to the documentation for an API.

Comments

@stefdoerr
Copy link

stefdoerr commented Nov 9, 2022 •
edited
Loading

Is your feature request related to a problem? Please describe.
I need to submit GPU jobs to google batch through the python API
Describe the solution you'd like
Make the InstancePolicy.accelerators option work
Describe alternatives you've considered
Not sure, I'd be open to suggestions

Accelerators seem to not be supported yet through the python API https://github.com/googleapis/python-batch/blob/main/google/cloud/batch_v1/types/job.py#L529-L531 even though they seem to be supported through the YAML/JSON config files https://cloud.google.com/batch/docs/create-run-job#gpu-container-job-api

I also looked at the RPC API and they are also not supported there.

Is there some sort of workaround to get it to work or do I have to give up on using the python API and switch to yaml files and the gcloud CLI tool (I'd prefer to avoid that since I'm writing a python library)?
@product-auto-label product-auto-label bot added the api: batch Issues related to the googleapis/python-batch API. label Nov 9, 2022
@parthea parthea added type: question Request for information or clarification. Not an issue. external This issue is blocked on a bug with the actual product. labels Nov 16, 2022
@parthea
Copy link
Contributor

parthea commented Nov 16, 2022 •
edited
Loading

Thanks for reporting this @stefdoerr! Googlers see b/258435497 for internal issue

@parthea parthea mentioned this issue Nov 19, 2022
Merged
1 task
@parthea parthea added type: docs Improvement to the documentation for an API. and removed type: question Request for information or clarification. Not an issue. labels Nov 19, 2022
@parthea
Copy link
Contributor

parthea commented Nov 19, 2022

The docs are not correct as the feature has been implemented. The issue with the docs will be fixed in #72

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees
No one assigned
Labels
api: batch Issues related to the googleapis/python-batch API. external This issue is blocked on a bug with the actual product. type: docs Improvement to the documentation for an API.
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

docs: fix minor docstring formatting googleapis/python-batch
2 participants
@parthea @stefdoerr