ludwig-ai--ludwig
wehub-resource-sync
593b94c120
pytest / Unit Tests (push) Has been cancelled
pytest / Integration (integration_tests_a) (push) Has been cancelled
pytest / Integration (integration_tests_b) (push) Has been cancelled
pytest / Integration (integration_tests_c) (push) Has been cancelled
pytest / Integration (integration_tests_d) (push) Has been cancelled
pytest / Integration (integration_tests_e) (push) Has been cancelled
pytest / Integration (integration_tests_f) (push) Has been cancelled
pytest / Integration (integration_tests_g) (push) Has been cancelled
pytest / Integration (integration_tests_h) (push) Has been cancelled
pytest / Integration (integration_tests_i) (push) Has been cancelled
pytest / Integration (integration_tests_j) (push) Has been cancelled
pytest / Distributed (distributed_a) (push) Has been cancelled
pytest / Distributed (distributed_b) (push) Has been cancelled
pytest / Distributed (distributed_c) (push) Has been cancelled
pytest / Distributed (distributed_d) (push) Has been cancelled
pytest / Distributed (distributed_e) (push) Has been cancelled
pytest / Distributed (distributed_f) (push) Has been cancelled
pytest / Minimal Install (push) Has been cancelled
pytest / Event File (push) Has been cancelled
pytest (slow) / py-slow (push) Has been cancelled
Publish JSON Schema / publish-schema (push) Has been cancelled
40 行
976 B
Docker
40 行
976 B
Docker
#
|
|
# Ludwig Docker image with full set of pre-requiste packages to support these capabilities
|
|
# text features
|
|
# image features
|
|
# audio features
|
|
# visualizations
|
|
# hyperparameter optimization
|
|
# distributed training
|
|
# model serving
|
|
#
|
|
|
|
FROM python:3.12-slim
|
|
|
|
RUN apt-get -y update && apt-get -y install \
|
|
git \
|
|
libsndfile1 \
|
|
build-essential \
|
|
g++ \
|
|
cmake \
|
|
ffmpeg \
|
|
sox \
|
|
libsox-dev
|
|
RUN pip install -U pip
|
|
|
|
ARG LUDWIG_VERSION
|
|
|
|
WORKDIR /ludwig
|
|
|
|
COPY . .
|
|
RUN if [ -n "${LUDWIG_VERSION}" ]; then \
|
|
pip install --no-cache-dir "ludwig[full]==${LUDWIG_VERSION}" --extra-index-url https://download.pytorch.org/whl/cpu; \
|
|
else \
|
|
pip install --no-cache-dir '.[full]' --extra-index-url https://download.pytorch.org/whl/cpu; \
|
|
fi
|
|
RUN pip install --no-cache-dir --force-reinstall torch==2.12.0 torchvision==0.27.0 torchaudio==2.11.0 --extra-index-url https://download.pytorch.org/whl/cpu
|
|
|
|
WORKDIR /data
|
|
|
|
ENTRYPOINT ["ludwig"]
|