Skip to content
Snippets Groups Projects
Commit 8ab8c333 authored by Jan Hartig's avatar Jan Hartig
Browse files

Add nvidia-smi to container for debugging issues

parent 571335e4
No related branches found
No related tags found
No related merge requests found
......@@ -6,6 +6,9 @@ RUN python3 -m pip --no-cache-dir install faster-whisper tqdm
# preload model
RUN python3 -c "import faster_whisper; faster_whisper.download_model('large-v2')"
# add nvidia-smi to container
RUN apt update && apt install -y nvidia-utils-515
COPY main.py .
VOLUME "/output/"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment