diff --git a/Dockerfile b/Dockerfile index 2aeb715..b21c00f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -5,7 +5,7 @@ RUN apt update && apt install -y --no-install-recommends \ && apt-get clean \ && rm -rf /var/lib/apt/lists/* -RUN python -m pip install --upgrade pip setuptools wheel +RUN pip install --upgrade pip setuptools wheel RUN pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu121 RUN pip install "unsloth[cu121-torch240] @ git+https://github.com/unslothai/unsloth.git"