CultriX commited on
Commit
3f3fe20
·
verified ·
1 Parent(s): 01d8b1a

Update dockerfile

Browse files
Files changed (1) hide show
  1. dockerfile +1 -1
dockerfile CHANGED
@@ -10,7 +10,7 @@ ADD . /opt/whisper-webui/
10
 
11
  # Latest version of transformers-pytorch-gpu seems to lack tk.
12
  # Further, pip install fails, so we must upgrade pip first.
13
- RUN apt-get -y install python3-tk
14
  RUN python3 -m pip install --upgrade pip
15
 
16
  RUN if [ "${WHISPER_IMPLEMENTATION}" = "whisper" ]; then \
 
10
 
11
  # Latest version of transformers-pytorch-gpu seems to lack tk.
12
  # Further, pip install fails, so we must upgrade pip first.
13
+ RUN apt-get -y install python3-tk nvidia-cudnn libcudnn-frontend-dev nvidia-cuda-dev
14
  RUN python3 -m pip install --upgrade pip
15
 
16
  RUN if [ "${WHISPER_IMPLEMENTATION}" = "whisper" ]; then \