Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add TPU support for neuro run/submit #982

Closed
dalazx opened this issue Aug 23, 2019 · 1 comment
Closed

Add TPU support for neuro run/submit #982

dalazx opened this issue Aug 23, 2019 · 1 comment
Assignees

Comments

@dalazx
Copy link
Contributor

dalazx commented Aug 23, 2019

The job payload changes were implemented here:https://github.com/neuromation/platform-api/pull/844/files

Also going to patch this part
https://github.com/neuromation/platform-api/blob/c45a965f1319a2f798a46d5c2d3e346ceb7792e7/platform_api/api.py#L63
to add an optional "tpu": {"type": "v3-8", "software_version": "1.14"}

neuro submit should expose two new options: --tpu-type and --tpu-sw-version.

@shagren
Copy link
Contributor

shagren commented Aug 30, 2019

closed via #985

@shagren shagren closed this as completed Aug 30, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

3 participants