-
Notifications
You must be signed in to change notification settings - Fork 1.4k
Add Python devcontainer with CUDA support #1176
base: main
Are you sure you want to change the base?
Conversation
Hey @Chuxel, can I get some feedback on the PR? |
Please, feedback on this would be appreciated! |
A few thoughts looking at this that partly caused some delay in response:
|
Hey @Chuxel What exactly is required for the refactor, any specific changes you have in mind? Feel free to request changes in a review. Node.js has been added for parity with the base python devcontainer, so a developer can simply switch to the cuda devcontainer without breaking existing code. Workloads such as data visualization can require node.js. |
@twsl your base.Dockerfile is working brilliantly, thanks! |
Adds a devcontainer based on the Python devcontainer with CUDA support.
Uses the nvidia/cuda image as base image.
Enables developers with focus on ML/DL to leverage devcontainers.
Could fix #180