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
44 lines
1.1 KiB
Docker
44 lines
1.1 KiB
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 nvidia/cuda:12.6.3-cudnn-devel-ubuntu24.04
|
|
|
|
RUN apt-get -y update && DEBIAN_FRONTEND="noninteractive" apt-get -y install \
|
|
python3.12 \
|
|
python3.12-venv \
|
|
python3.12-dev \
|
|
git \
|
|
libsndfile1 \
|
|
cmake \
|
|
ffmpeg \
|
|
sox \
|
|
libsox-dev
|
|
|
|
RUN python3.12 -m venv /opt/ludwig-venv
|
|
ENV PATH="/opt/ludwig-venv/bin:$PATH"
|
|
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/cu126; \
|
|
else \
|
|
pip install --no-cache-dir '.[full]' --extra-index-url https://download.pytorch.org/whl/cu126; \
|
|
fi
|
|
RUN pip install --no-cache-dir --force-reinstall torch==2.12.0 torchvision==0.27.0 torchaudio==2.11.0 --index-url https://download.pytorch.org/whl/cu126
|
|
|
|
WORKDIR /data
|
|
|
|
ENTRYPOINT ["ludwig"]
|