From fa7e659eb3b063b4267d8aeb70ea0abe38a65b6c Mon Sep 17 00:00:00 2001
From: Rayed Bin Wahed <rayed@therapservices.net>
Date: Mon, 26 Feb 2024 20:24:15 +0800
Subject: [PATCH] Add devcontainer

---
 .devcontainer/Dockerfile        | 12 ++++++++++++
 .devcontainer/devcontainer.json | 19 +++++++++++++++++++
 2 files changed, 31 insertions(+)
 create mode 100644 .devcontainer/Dockerfile
 create mode 100644 .devcontainer/devcontainer.json

diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile
new file mode 100644
index 0000000..c8edb94
--- /dev/null
+++ b/.devcontainer/Dockerfile
@@ -0,0 +1,12 @@
+FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-devel
+
+RUN <<EOF
+apt-get update
+apt-get install -y --no-install-recommends \
+    git
+rm -rf /var/lib/apt/lists/*
+EOF
+
+COPY requirements.txt requirements.txt
+
+RUN pip install --no-cache-dir -r requirements.txt
diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
new file mode 100644
index 0000000..52cde09
--- /dev/null
+++ b/.devcontainer/devcontainer.json
@@ -0,0 +1,19 @@
+{
+  "name": "LLMs From Scratch",
+  "build": {
+    "context": "..",
+    "dockerfile": "Dockerfile"
+  },
+  "runArgs": ["--runtime=nvidia", "--gpus=all"],
+  "customizations": {
+    "vscode": {
+      "extensions": [
+        "ms-python.python",
+        "ms-python.black-formatter",
+        "ms-azuretools.vscode-docker",
+        "ms-toolsai.jupyter",
+        "esbenp.prettier-vscode"
+      ]
+    }
+  }
+}