mirror of
https://github.com/TaterTotterson/microWakeWord-Trainer-Nvidia-Docker.git
synced 2026-06-12 20:10:19 -06:00
47 lines
1.7 KiB
Plaintext
47 lines
1.7 KiB
Plaintext
# CUDA 12.6 + cuDNN devel (Ubuntu 22.04)
|
|
FROM nvidia/cuda:12.6.2-cudnn-devel-ubuntu22.04
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive \
|
|
PYTHONUNBUFFERED=1 \
|
|
PIP_NO_CACHE_DIR=1 \
|
|
PIP_ROOT_USER_ACTION=ignore \
|
|
HF_HUB_DISABLE_SYMLINKS_WARNING=1 \
|
|
XLA_FLAGS="--xla_gpu_cuda_data_dir=/usr/local/cuda" \
|
|
PATH="/usr/local/cuda/bin:${PATH}" \
|
|
LD_LIBRARY_PATH="/usr/local/cuda/lib64:${LD_LIBRARY_PATH}"
|
|
|
|
# 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 git-lfs \
|
|
build-essential g++ cmake \
|
|
libsndfile1 libsndfile1-dev libffi-dev \
|
|
ffmpeg \
|
|
&& 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
|
|
|
|
# ---- No cuDNN repo meddling needed if using TF 2.17.x ----
|
|
|
|
# Python deps
|
|
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
|
|
|
|
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"] |