-
Notifications
You must be signed in to change notification settings - Fork 1
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
Labels
enhancement
New feature or request
Comments
8eba477 introduced rudimentary support. This could be a bit cleaner though, i.e. no duplication in Dockerfile-core* |
Now the question is instrumentation, i.e running the containers with
TL&DR: Use GPU by default if possible, use CPU if told by the user |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
We current run the workflow in a GPU-less container. From what I have tested, this could get GPU support by:
FROM nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04
, anddocker run --gpus all
It should still work on CPU-only systems.
The text was updated successfully, but these errors were encountered: