Update whisper_pipeline_cu118/dockerfile

#1
Files changed (1) hide show
  1. whisper_pipeline_cu118/dockerfile +2 -1
whisper_pipeline_cu118/dockerfile CHANGED
@@ -1,5 +1,5 @@
1
  # Use nvidia/cuda as base image with Python
2
- FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
3
 
4
  # Use args
5
  ARG USE_CUDA
@@ -14,6 +14,7 @@ ENV ENV=prod \
14
  # Install GCC and build tools
15
  RUN apt-get update && \
16
  apt-get install -y gcc build-essential curl git pkg-config libicu-dev && \
 
17
  apt-get clean && \
18
  rm -rf /var/lib/apt/lists/*
19
 
 
1
  # Use nvidia/cuda as base image with Python
2
+ FROM nvidia/cuda:12.0.0-cudnn8-runtime-ubuntu22.04
3
 
4
  # Use args
5
  ARG USE_CUDA
 
14
  # Install GCC and build tools
15
  RUN apt-get update && \
16
  apt-get install -y gcc build-essential curl git pkg-config libicu-dev && \
17
+ apt-get install -y --no-install-recommends ffmpeg libsm6 libxext6 \ #for ffmpeg preprocessing
18
  apt-get clean && \
19
  rm -rf /var/lib/apt/lists/*
20