Skip to content

Commit

Permalink
added dockerfile. pip install is very temperamental about install ord…
Browse files Browse the repository at this point in the history
…er of dependencies
  • Loading branch information
dsingal0 committed Jul 6, 2024
1 parent 9b4cc93 commit 29ee45b
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -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"

0 comments on commit 29ee45b

Please sign in to comment.