Added Dockerfile

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"

README.md

@@ -170,6 +170,17 @@ pip install "unsloth[cu121-ampere-torch230] @ git+https://github.com/unslothai/u
 nvcc
 python -m xformers.info
 python -m bitsandbytes
+```  
+
+### Docker Installation
+1. Edit Dockerfile to match required CUDA and Pytorch version. By default, CUDA 12.1.1 and Pytorch 2.3 are used, but you modify to use CUDA 11.8 and Pytorch 2.x instead as well as install optional dependencies by changing `unsloth[cu121-torch230]`.  
+2. Build docker image
+```bash
+docker build -t unsloth:latest .
+```
+3. Run unsloth docker image
+```bash
+docker run --gpus all -it --rm -v $PWD:/workspace --ulimit memlock=-1 --ulimit stack=67108864 --ipc host unsloth:latest
 ```
 
 ## 📜 Documentation