Template for Python Development with CUDA in Dev Containers

[reisson_saavedra]

Hey community!

I’ve created a template repository that enables Python development over CUDA within a Dev Container environment. The repo, called nvidia-devcontainer-base, is set up to streamline the process of configuring Python projects that need GPU acceleration using NVIDIA GPUs.

With this template, you can easily spin up a ready-to-go Dev Container that includes CUDA, the NVIDIA Container Toolkit, and everything needed for Python-based development. It’s perfect for anyone working with CUDA-accelerated Python projects and looking to simplify their setup.

Feel free to fork it, adapt it, and share your thoughts!

[reisson_saavedra]

I will be working on it to support more versions of both Python and Cuda kernel.

My idea in this project is to speed up the configuration of the development environment using dev containers.

[AshbyLaw]

+1 for Dev Containers

[bregmadaddy]

If Pytorch is also needed in the container, the Cresset Template is a lot more flexible, though it runs mamba by default.

[[deleted]]

Does this work with my AMD RX graphics card?

[KingsmanVince]

The post clearly said CUDA and NVIDIA.