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)
FROM nvidia/cuda:12.6.2-cudnn-devel-ubuntu22.04
FROM nvidia/cuda:12.8.1-cudnn-devel-ubuntu22.04
ENV DEBIAN_FRONTEND=noninteractive \
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
RUN pip install --upgrade pip \
&& 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
RUN mkdir -p /data