22 lines
673 B
Docker
22 lines
673 B
Docker
FROM ubuntu:22.04 AS builder
|
|
|
|
# Установка зависимостей
|
|
|
|
RUN apt update && apt-get install -y \
|
|
python3 pip git curl wget nano sudo apt-utils && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# Установка RKLLM Toolkit
|
|
RUN git clone https://github.com/airockchip/rknn-llm
|
|
|
|
# Установка Python-пакета
|
|
RUN pip3 install /ezrknn-llm/rkllm-toolkit/packages/rkllm_toolkit-1.2.3-cp312-cp312-linux_x86_64.whl
|
|
|
|
# Клонирование модели
|
|
WORKDIR /models
|
|
|
|
# Копируем скрипт компиляции
|
|
COPY entrypoint.py /entrypoint.py
|
|
|
|
# Точка входа для компиляции
|
|
ENTRYPOINT ["python3", "/entrypoint.py"] |