Add Devcontainer for VScode.

.devcontainer/devcontainer.json

@@ -0,0 +1,44 @@
+// For format details, see https://aka.ms/devcontainer.json. For config options, see the
+// README at: https://github.com/devcontainers/templates/tree/main/src/ubuntu
+{
+	"name": "",
+	// Or use a Dockerfile or Docker Compose file. More info: https://containers.dev/guide/dockerfile
+	//"image": "mcr.microsoft.com/devcontainers/base:jammy"
+	"build": {
+		"dockerfile": "../Dockerfile"
+	},
+	// Features to add to the dev container. More info: https://containers.dev/features.
+	// "features": {},
+	// Use 'forwardPorts' to make a list of ports inside the container available locally.
+	// "forwardPorts": [],
+	// Use 'postCreateCommand' to run commands after the container is created.
+	// "postCreateCommand": "uname -a",
+	// Configure tool-specific properties.
+	"customizations": {
+		"vscode": {
+			"extensions": [
+				"ms-python.python",
+				"ms-python.vscode-pylance",
+				"ms-toolsai.jupyter",
+				"ms-toolsai.vscode-jupyter-cell-tags",
+				"ms-toolsai.tensorboard",
+				"ms-toolsai.jupyter-keymap",
+				"ms-toolsai.jupyter-renderers",
+				"ms-toolsai.vscode-jupyter-slideshow",
+				"ms-toolsai.datawrangler",
+				"ms-azuretools.vscode-docker",
+				"github.vscode-github-actions",
+				"github.copilot",
+				"github.copilot-chat",
+				"mechatroner.rainbow-csv"
+			]
+		}
+	},
+	"runArgs": [
+		"--shm-size=1gb",
+		"--gpus",
+		"all"
+	]
+	// Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root.
+	// "remoteUser": "root"
+}

Dockerfile

@@ -0,0 +1,22 @@
+FROM ubuntu:22.04
+
+ENV PIP_ROOT_USER_ACTION=ignore
+WORKDIR /tmp/
+COPY requirements.txt packages.txt ./
+
+RUN apt-get -y update && apt-get -y upgrade
+RUN xargs apt -y install < packages.txt
+
+RUN wget -v https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb && dpkg -i cuda-keyring_1.1-1_all.deb && apt-get -y update && apt-get -y install cuda-toolkit-12-8
+
+RUN wget -v https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb && dpkg -i cuda-keyring_1.1-1_all.deb && apt-get -y update && apt-get -y install cudnn
+
+RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
+
+RUN rm -f /usr/lib/python3.12/EXTERNALLY-MANAGED
+RUN pip3 install --no-cache-dir -r requirements.txt
+
+RUN apt-get clean
+RUN rm -rf /tmp/*
+
+CMD ["bash"]