From 43260eb013609fce90685c59866ddbbb5c96bcdf Mon Sep 17 00:00:00 2001 From: marco Date: Sat, 26 Apr 2025 12:11:13 +0200 Subject: [PATCH] fix base image to run on Container OS update README --- Dockerfile | 2 +- Makefile | 6 +++++- README.md | 18 ++++++++++++++++++ 3 files changed, 24 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 4410442..b11f25c 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -FROM pytorch/pytorch:2.4.0-cuda11.8-cudnn9-runtime +FROM pytorch/pytorch:2.7.0-cuda12.6-cudnn9-runtime WORKDIR /app RUN mkdir /data diff --git a/Makefile b/Makefile index 2537699..a08258a 100644 --- a/Makefile +++ b/Makefile @@ -20,4 +20,8 @@ req: setup: python3 -m venv .venv - ./.venv/bin/pip3 install -r requirements.txt \ No newline at end of file + ./.venv/bin/pip3 install -r requirements.txt + +push: + docker tag sermas/speechbrain ghcr.io/sermas-eu/speechbrain:test + docker push ghcr.io/sermas-eu/speechbrain:test diff --git a/README.md b/README.md index 284e2ca..2f536e1 100644 --- a/README.md +++ b/README.md @@ -1,2 +1,20 @@ # speechbrain SERMAS speechbrain service + + +### How to run on Google Container OS VM +``` +sudo cos-extensions install gpu +sudo mount --bind /var/lib/nvidia /var/lib/nvidia +sudo mount -o remount,exec /var/lib/nvidia +/var/lib/nvidia/bin/nvidia-smi + +docker run \ + --volume /var/lib/nvidia/lib64:/usr/local/nvidia/lib64 \ + --volume /var/lib/nvidia/bin:/usr/local/nvidia/bin \ + --device /dev/nvidia0:/dev/nvidia0 \ + --device /dev/nvidia-uvm:/dev/nvidia-uvm \ + --device /dev/nvidiactl:/dev/nvidiactl \ +-p 80:5011 \ + ghcr.io/sermas-eu/speechbrain:test + ```