-
Notifications
You must be signed in to change notification settings - Fork 416
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 new NVIDIA CUDA feature #80
Conversation
There's a bit of a gap in the dev container spec for features that is needed here. Specifically, there's no way that I'm aware of for a feature to specify GPU usage - which I'd expect you'd want for this feature, correct? Is there any reason to use this feature without it? EDIT: Raised devcontainers/spec#82 |
Not really. These libraries enable tools to leverage NVIDIA GPUs. That said, you can successfully install this feature without a GPU, but it won't be too useful. |
This PR is ready for another set of eyes! Here are my notable changes:
|
… jungaretti/nvidia-feature
Adds a new feature that installs Nvidia's CUDA and cuDNN libraries. These libraries enable tools such as TensorFlow to leverage an Nvidia GPU to perform highly-parallelizable tasks.
Motivation
This new feature would dramatically decrease friction for developers who use Codespaces to work on machine learning projects. Instead of copying, pasting, and executing this script, developers could simply add the new
nvidia
feature to theirdevcontainer.json
and rebuild their codespace.Changes