diff --git a/Dockerfile b/Dockerfile
index 0db592b..d3767bb 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,8 +1,8 @@
-FROM nvidia/cuda:11.7.1-base-ubuntu22.04
+FROM nvidia/cuda:12.4.1-devel-ubuntu22.04
 
 # Install Whisper
 WORKDIR /usr/src
-ARG WHISPER_VERSION='1.0.1'
+ARG WHISPER_VERSION='2.0.0'
 
 RUN \
     apt-get update \