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
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 \
&& 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 \
@@ -40,7 +42,6 @@ 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"]