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

Improve CUDA/XLA setup and CPU fallback

Browse Source
This commit is contained in:
Tater Totterson
2026-01-20 20:47:58 -06:00
parent e72773f0ef
commit 8669ccdfd4
4 changed files with 38 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dockerfile
View File

@@ -6,7 +6,7 @@ ENV DEBIAN_FRONTEND=noninteractive
# System deps
RUN apt-get update && apt-get install -y --no-install-recommends \
python3.12 python3.12-venv python3.12-dev python3-pip python-is-python3 \
git wget curl unzip ca-certificates nano less \
git wget curl unzip ca-certificates nano less nvidia-cuda-toolkit \
&& rm -rf /var/lib/apt/lists/* \
&& mkdir -p /data