diff --git a/Dockerfile b/Dockerfile
index 820ae94a..bc3a3612 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.4.1-base-ubuntu22.04
+FROM nvidia/cuda:12.8.1-base-ubuntu22.04
 ENV DEBIAN_FRONTEND noninteractive
 ENV CMDARGS --listen
 
diff --git a/requirements_docker.txt b/requirements_docker.txt
index 21883adf..dde4431a 100644
--- a/requirements_docker.txt
+++ b/requirements_docker.txt
@@ -1,2 +1,2 @@
-torch==2.1.0
-torchvision==0.16.0
+torch==2.8.0
+torchvision==0.23.0