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-12-05 07:44:13 -06:00
committed by GitHub
parent efcf8a4add
commit 7c9c6f7a88
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
dockerfile
View File

@@ -1,5 +1,5 @@
# CUDA 12.6 + cuDNN devel (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.8.1-cudnn-devel-ubuntu22.04
ENV DEBIAN_FRONTEND=noninteractive \ ENV DEBIAN_FRONTEND=noninteractive \
PYTHONUNBUFFERED=1 \ PYTHONUNBUFFERED=1 \
@@ -29,7 +29,11 @@ RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1 \
COPY requirements.txt /tmp/requirements.txt COPY requirements.txt /tmp/requirements.txt
RUN pip install --upgrade pip \ RUN pip install --upgrade pip \
&& pip install "numpy==1.26.4" "cython>=0.29.36" \ && pip install "numpy==1.26.4" "cython>=0.29.36" \
&& pip install -r /tmp/requirements.txt && pip install -r /tmp/requirements.txt \
&& pip install \
torch==2.7.1 \
torchaudio==2.7.1 \
--index-url https://download.pytorch.org/whl/cu128
# Workspace + notebook fallback # Workspace + notebook fallback
RUN mkdir -p /data RUN mkdir -p /data

2
requirements.txt
View File

@@ -1,6 +1,4 @@
# --- Core training (Microwakeword) --- # --- Core training (Microwakeword) ---
torch==2.5.1
torchaudio==2.5.1
# TensorFlow stack (match CUDA/cuDNN in the base image) # TensorFlow stack (match CUDA/cuDNN in the base image)
tensorflow==2.17.1 tensorflow==2.17.1