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

GPU support #6

Open
mikegerber opened this issue Feb 10, 2020 · 4 comments
Open

GPU support #6

mikegerber opened this issue Feb 10, 2020 · 4 comments
Assignees
Labels
enhancement New feature or request

Comments

@mikegerber
Copy link
Member

We current run the workflow in a GPU-less container. From what I have tested, this could get GPU support by:

  • basing on a CUDA image, e.g. FROM nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04, and
  • running docker run --gpus all

It should still work on CPU-only systems.

@mikegerber mikegerber added the enhancement New feature or request label Feb 10, 2020
@mikegerber mikegerber self-assigned this Feb 10, 2020
mikegerber added a commit that referenced this issue Feb 10, 2020
@mikegerber
Copy link
Member Author

8eba477 introduced rudimentary support. This could be a bit cleaner though, i.e. no duplication in Dockerfile-core*

@mikegerber
Copy link
Member Author

mikegerber commented Apr 23, 2021

Now the question is instrumentation, i.e running the containers with docker run --gpus=all.

  • Run with --gpus=all if possible
    • (On CPU-only systems this fails, so we need a way to detect the availability of nvidia-container-runtime)
  • Use CUDA_VISIBLE_DEVICES to support disabling running on GPU
    • We need that sometimes to make use of the CPU ressources available

TL&DR: Use GPU by default if possible, use CPU if told by the user

@mikegerber
Copy link
Member Author

@mikegerber
Copy link
Member Author

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

No branches or pull requests

1 participant