CUDA Development Container

This repo configures a CUDA enabled python development container. If you are unfamiliar with development containers, check out this page.

Installation

1. Add repo as submodule

git submodule add git@github.com:raphaelschwinger/cuda-python-devcontainer.git .devcontainer

2. Set CUDA_VERSION and VARIANT in .devcontainer/devcontainer.json to your desired values. The CUDA and Ubuntu version has to match with image tags found here.

3. Reopen in container

4. Init poetry project

poetry init

5. Start installing packages

poetry add <package>

6. Start virtualenv

eval $(poetry env activate)