diff --git a/Dockerfile b/Dockerfile
index 22d787f..d0815b9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.8.1-runtime-ubuntu22.04
+FROM docker.io/nvidia/cuda:12.8.1-runtime-ubuntu22.04
 
 ARG RUNTIME=nvidia
 
diff --git a/Dockerfile.cu128 b/Dockerfile.cu128
index 40e849c..a62618e 100644
--- a/Dockerfile.cu128
+++ b/Dockerfile.cu128
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.8.1-runtime-ubuntu22.04
+FROM docker.io/nvidia/cuda:12.8.1-runtime-ubuntu22.04
 
 ARG RUNTIME=nvidia
 
diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 1811056..68b552e 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -1,4 +1,4 @@
-FROM rocm/dev-ubuntu-22.04:latest
+FROM docker.io/rocm/dev-ubuntu-22.04:latest
 
 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE=1