blmdsydm commited on
Commit
0e62583
1 Parent(s): e40ee50

Update dockerfile

Browse files

change base image to nvidia

Files changed (1) hide show
  1. dockerfile +1 -1
dockerfile CHANGED
@@ -1,6 +1,6 @@
1
  # docker build -t whisper-webui --build-arg WHISPER_IMPLEMENTATION=whisper .
2
 
3
- FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
4
  EXPOSE 7860
5
 
6
  ARG WHISPER_IMPLEMENTATION=faster-whisper
 
1
  # docker build -t whisper-webui --build-arg WHISPER_IMPLEMENTATION=whisper .
2
 
3
+ FROM nvidia/cuda:11.7.0-cudnn8-runtime-ubuntu18.04
4
  EXPOSE 7860
5
 
6
  ARG WHISPER_IMPLEMENTATION=faster-whisper