From 57d192d62bf48198d12c62414980e03002c527b3 Mon Sep 17 00:00:00 2001
From: Akemi Izuko <akemi@noway.moe>
Date: Mon, 11 Nov 2024 23:54:14 -0700
Subject: [PATCH] Add dockerfile

---
 Dockerfile | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 Dockerfile

diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..f4a8c59
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,18 @@
+FROM docker.io/python:3.12.7-bookworm
+
+WORKDIR /workspace
+
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    git \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN pip install --no-cache-dir \
+    jax[cuda12] \
+    equinox \
+    torch \
+    optax \
+    torchvision
+
+CMD ["/usr/bin/bash"]