13 lines
575 B
Docker
13 lines
575 B
Docker
ARG BUILD_TAG=latest
|
|
ARG BUILD_BASE_TAG=latest
|
|
FROM gitea.simcop2387.info/simcop2387/text-gen-os-base:$BUILD_BASE_TAG
|
|
|
|
RUN mkdir /app/
|
|
COPY ./unsloth/ /app/
|
|
WORKDIR /app/
|
|
RUN python3 -m venv /app/venv/
|
|
RUN --mount=type=cache,target=/root/.cache/pip,rw . /app/venv/bin/activate && \
|
|
pip3 install --upgrade pip setuptools wheel packaging && \
|
|
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
|
|
RUN --mount=type=cache,target=/root/.cache/pip,rw . /app/venv/bin/activate && \
|
|
/app/venv/bin/pip3 install "/app/[cu121-ampere]"
|