taco/microWakeWord-Trainer-Nvidia-Docker
1
0
Fork 0
mirror of https://github.com/TaterTotterson/microWakeWord-Trainer-Nvidia-Docker.git synced 2026-06-12 20:10:19 -06:00
Code Issues Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
Tater Totterson
2025-09-27 19:44:55 -05:00
committed by GitHub
parent 901097d60e
commit c6fe612143
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
dockerfile
View File

@@ -1,4 +1,4 @@
# CUDA + cuDNN userspace from NVIDIA (Ubuntu 22.04) # CUDA 12.6 + cuDNN devel (Ubuntu 22.04)
FROM nvidia/cuda:12.6.2-cudnn-devel-ubuntu22.04 FROM nvidia/cuda:12.6.2-cudnn-devel-ubuntu22.04
ENV DEBIAN_FRONTEND=noninteractive \ ENV DEBIAN_FRONTEND=noninteractive \
@@ -23,6 +23,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1 \ RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1 \
&& update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 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 # Python deps
COPY requirements.txt /tmp/requirements.txt COPY requirements.txt /tmp/requirements.txt
RUN pip install --upgrade pip \ RUN pip install --upgrade pip \
@@ -40,7 +42,6 @@ RUN chmod +x /usr/local/bin/startup.sh
EXPOSE 8888 EXPOSE 8888
# Launch JupyterLab (tokenless for local dev; set a token if you want auth)
CMD ["/bin/bash", "-lc", "/usr/local/bin/startup.sh && \ CMD ["/bin/bash", "-lc", "/usr/local/bin/startup.sh && \
exec jupyter lab --ip=0.0.0.0 --port=8888 --no-browser --allow-root \ exec jupyter lab --ip=0.0.0.0 --port=8888 --no-browser --allow-root \
--ServerApp.token='' --ServerApp.password='' --ServerApp.root_dir=/data"] --ServerApp.token='' --ServerApp.password='' --ServerApp.root_dir=/data"]

9
requirements.txt
View File

@@ -1,11 +1,14 @@
# --- Core training (Microwakeword) --- # --- Core training (Microwakeword) ---
torch==2.5.1 torch==2.5.1
torchaudio==2.5.1 torchaudio==2.5.1
tensorboard==2.18.0
# TensorFlow stack (match CUDA/cuDNN in the base image)
tensorflow==2.17.1
tensorboard==2.17.1
tensorboard-data-server==0.7.2 tensorboard-data-server==0.7.2
tensorflow==2.18.0
tensorflow-estimator==2.13.0
tensorflow-io-gcs-filesystem==0.37.1 tensorflow-io-gcs-filesystem==0.37.1
# tensorflow-estimator not needed (remove)
numpy==1.26.4 numpy==1.26.4
scipy==1.12.0 scipy==1.12.0
librosa==0.10.2.post1 librosa==0.10.2.post1