Skip to content

Accelerator support #10

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

Open
rivershah opened this issue Nov 17, 2022 · 0 comments
Open

Accelerator support #10

rivershah opened this issue Nov 17, 2022 · 0 comments

Comments

@rivershah
Copy link

rivershah commented Nov 17, 2022

Feature request to expose gpu support for ci_workers. For unit testing am requiring GPUs. Use case is verification and testing of machine learning code and cpu only machines is not an option as behaviour changes sufficiently to require testing on target devices. It is not immediately obvious how gpu accelerators can be used with ci_workers. The required machine configs for the ci_workers are:

machine_type = "n1-standard-8"
guest_accelerator {
  type  = "nvidia-tesla-t4"
  count = 2
}

machine_type = "a2-highgpu-4g"
guest_accelerator {
  type  = "nvidia-tesla-a100"
  count = 4
}

Sorry I don't have the skill set to fully understand how the machine image, nvidia driver loading and any cuda dependencies need to work with the gitlab runner or would have submitted a PR

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant