diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json new file mode 100644 index 000000000..84f99bf49 --- /dev/null +++ b/.devcontainer/devcontainer.json @@ -0,0 +1,13 @@ +{ + "name": "NerfStudio Dev Container", + "dockerComposeFile": "docker-compose.yml", + "service": "nerfstudio", + "workspaceFolder": "/workspace", + "shutdownAction": "stopCompose", + "extensions": [ + "mutantdino.resourcemonitor", + "ms-azuretools.vscode-docker", + "nvidia.nsight-vscode-edition", + "ms-python.python" + ] +} \ No newline at end of file diff --git a/.devcontainer/docker-compose.yml b/.devcontainer/docker-compose.yml new file mode 100644 index 000000000..6376422cf --- /dev/null +++ b/.devcontainer/docker-compose.yml @@ -0,0 +1,28 @@ +version: "3.8" + +services: + nerfstudio: + shm_size: '12gb' + image: dromni/nerfstudio:0.1.17 + # build: + # context: .. + # dockerfile: ./Dockerfile + stdin_open: true + tty: true + environment: + NVIDIA_DRIVER_CAPABILITIES: compute,utility,graphics + DISPLAY: $DISPLAY + ports: + - "7007:7007" + volumes: + - ../:/workspace/ + - /tmp/.X11-unix:/tmp/.X11-unix + working_dir: /workspace/ + command: /bin/bash + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: 1 + capabilities: [ gpu ] \ No newline at end of file