Skip to content

Commit

Permalink
Added devcontainer for cuda development
Browse files Browse the repository at this point in the history
  • Loading branch information
perara committed Jun 21, 2024
1 parent 1d833ce commit d26cfee
Show file tree
Hide file tree
Showing 3 changed files with 41 additions and 0 deletions.
4 changes: 4 additions & 0 deletions .devcontainer/cuda/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
FROM nvcr.io/nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04

# Run GIT
RUN apt-get update && apt-get install -y git
13 changes: 13 additions & 0 deletions .devcontainer/cuda/devcontainer.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
{
"name": "TMU Development Environment",
"dockerComposeFile": "docker-compose.yml",
"service": "devcontainer",
"workspaceFolder": "/workspace",
"customizations": {
"vscode": {
"extensions": [
"ms-python.python"
]
}
}
}
24 changes: 24 additions & 0 deletions .devcontainer/cuda/docker-compose.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
services:
devcontainer:
networks:
- internal
build:
context: ../../
dockerfile: .devcontainer/cuda/Dockerfile
volumes:
- ./:/workspace
- /var/run/docker.sock:/var/run/docker.sock
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [gpu]

networks:
internal:
driver: bridge

volumes:
config:

0 comments on commit d26cfee

Please sign in to comment.