mirror of
https://github.com/TaterTotterson/microWakeWord-Trainer-Nvidia-Docker.git
synced 2026-06-12 20:10:19 -06:00
41 lines
1.5 KiB
Plaintext
41 lines
1.5 KiB
Plaintext
# CUDA + cuDNN userspace from NVIDIA (no manual repo installs needed)
|
|
FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive \
|
|
PYTHONUNBUFFERED=1 \
|
|
PIP_NO_CACHE_DIR=1
|
|
|
|
# System deps (+dev headers for building C/C++ extensions)
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
python3.10 python3.10-venv python3.10-distutils python3.10-dev python3-pip \
|
|
git wget curl unzip ca-certificates \
|
|
build-essential g++ cmake \
|
|
libsndfile1 libsndfile1-dev libffi-dev \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Use python3.10 everywhere
|
|
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1 \
|
|
&& update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
|
|
|
|
# Python deps
|
|
# Pre-install numpy (and cython) so native builds (webrtcvad, pymicro_features) have headers ready
|
|
COPY requirements.txt /tmp/requirements.txt
|
|
RUN pip install --upgrade pip \
|
|
&& pip install "numpy==1.26.4" "cython>=0.29.36" \
|
|
&& pip install -r /tmp/requirements.txt
|
|
|
|
# Workspace + notebook fallback
|
|
RUN mkdir -p /data
|
|
WORKDIR /data
|
|
COPY microWakeWord_training_notebook.ipynb /root/
|
|
|
|
# Startup script (copies default notebook if missing)
|
|
COPY startup.sh /usr/local/bin/startup.sh
|
|
RUN chmod +x /usr/local/bin/startup.sh
|
|
|
|
EXPOSE 8888
|
|
|
|
# Launch JupyterLab (tokenless for local dev; set a token if you want auth)
|
|
CMD ["/bin/bash", "-lc", "/usr/local/bin/startup.sh && \
|
|
exec jupyter lab --ip=0.0.0.0 --port=8888 --no-browser --allow-root \
|
|
--ServerApp.token='' --ServerApp.password='' --ServerApp.root_dir=/data"] |