From 29ee45b7f50b9ce6ce63c234d5b111b8682b24b0 Mon Sep 17 00:00:00 2001
From: Dhruv Singal <dhruvsingalabc@gmail.com>
Date: Sat, 6 Jul 2024 15:49:42 -0700
Subject: [PATCH] added dockerfile. pip install is very temperamental about
 install order of dependencies

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

diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 000000000..5aba4b134
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,13 @@
+
+ARG CUDA_VERSION=12.1.1
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-cudnn8-devel-ubuntu22.04
+ENV DEBIAN_FRONTEND=noninteractive
+RUN echo 'tzdata tzdata/Areas select America' | debconf-set-selections \
+    && echo 'tzdata tzdata/Zones/America select Los_Angeles' | debconf-set-selections \
+    && apt-get update -y \
+    && apt-get install -y software-properties-common python3 python3-pip  python-is-python3 git curl sudo
+RUN pip install --upgrade pip
+RUN pip install --upgrade torch triton packaging
+RUN pip install --upgrade --force-reinstall --no-cache-dir \
+    ninja einops flash-attn xformers trl peft accelerate bitsandbytes \
+    "unsloth[cu121-torch230] @ git+https://github.com/unslothai/unsloth.git"