Main building pieces
This commit is contained in:
parent
2bbe23c1b6
commit
3ce7ec05fb
2 changed files with 43 additions and 1 deletions
34
.woodpecker/build-containers.yaml
Normal file
34
.woodpecker/build-containers.yaml
Normal 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
|
10
Dockerfile
10
Dockerfile
|
@ -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]"
|
||||||
|
|
Loading…
Add table
Reference in a new issue