Main building pieces

This commit is contained in:
Ryan Voots 2024-07-13 11:32:59 -04:00
parent 2bbe23c1b6
commit 3ce7ec05fb
2 changed files with 43 additions and 1 deletions

View file

@ -0,0 +1,34 @@
when:
- event: manual
- event: push
steps:
- name: build_gpu
image: woodpeckerci/plugin-docker-buildx
settings:
build_args:
- "torch_cuda_arch_list=6.0 6.1 8.0 8.6+PTX"
repo: gitea.simcop2387.info/simcop2387/unsloth-base-image
auto_tag: true
tags:
- "build-${CI_PIPELINE_NUMBER}"
logins:
- registry: https://gitea.simcop2387.info/
username: simcop2387
password:
from_secret: docker_token
# - name: build_cpu
# image: woodpeckerci/plugin-docker-buildx
# settings:
# repo: gitea.simcop2387.info/simcop2387/vllm-base-cpu-image
# context: vllm-mirror/
# dockerfile: vllm-mirror/Dockerfile.cpu
# auto_tag: true
# tags:
# - "build-${CI_PIPELINE_NUMBER}"
# logins:
# - registry: https://gitea.simcop2387.info/
# username: simcop2387
# password:
# from_secret: docker_token

View file

@ -2,4 +2,12 @@ ARG BUILD_TAG=latest
ARG BUILD_BASE_TAG=latest ARG BUILD_BASE_TAG=latest
FROM gitea.simcop2387.info/simcop2387/text-gen-os-base:$BUILD_BASE_TAG 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]"