Update dockerfile

install cudnn in dockerfile

dockerfile

@@ -1,6 +1,6 @@
 # docker build -t whisper-webui --build-arg WHISPER_IMPLEMENTATION=whisper .
 
-FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
+FROM huggingface/transformers-pytorch-gpu
 EXPOSE 7860
 
 ARG WHISPER_IMPLEMENTATION=faster-whisper
@@ -12,6 +12,8 @@ ADD . /opt/whisper-webui/
 # Further, pip install fails, so we must upgrade pip first.
 RUN apt-get -y install python3-tk
 RUN  python3 -m pip install --upgrade pip
+RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcudnn8_8.5.0.96-1+cuda11.7_amd64.deb
+RUN dpkg -i libcudnn8_8.5.0.96-1+cuda11.7_amd64.deb && rm libcudnn8_8.5.0.96-1+cuda11.7_amd64.deb
 
 RUN if [ "${WHISPER_IMPLEMENTATION}" = "whisper" ]; then \
     python3 -m pip install -r /opt/whisper-webui/requirements-whisper.txt; \