diff --git a/Dockerfile b/Dockerfile
index d4844c7..b1c3358 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-ARG CUDA_VERSION=12.1
+ARG CUDA_VERSION=12.1.0
 ARG PYTHON_VERSION=3.10
 ARG GET_PIP_URL="https://bootstrap.pypa.io/get-pip.py"