This repo configures a CUDA enabled python development container. If you are unfamiliar with development containers, check out this page.
- Add repo as submodule
git submodule add git@github.com:raphaelschwinger/cuda-python-devcontainer.git .devcontainer
-
Set
CUDA_VERSIONandVARIANTin.devcontainer/devcontainer.jsonto your desired values. The CUDA and Ubuntu version has to match with image tags found here. -
Reopen in container
-
Init poetry project
poetry init
- Start installing packages
poetry add <package>
- Start virtualenv
eval $(poetry env activate)