diff --git a/docker/Dockerfile.x64 b/docker/Dockerfile.x64
index e421f90..2fd735c 100644
--- a/docker/Dockerfile.x64
+++ b/docker/Dockerfile.x64
@@ -3,7 +3,7 @@
 # ─────────────────────────────────────────────────────────────────────────────
 # Stage 1 – base: CUDA + ROS2 + locale/timezone
 # ─────────────────────────────────────────────────────────────────────────────
-FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 AS base
+FROM nvidia/cuda:12.6.3-cudnn-runtime-ubuntu22.04  AS base
 
 LABEL description="ROS2 Humble + CUDA – elevation_mapping_cupy baked in"
 LABEL version="2.0"
@@ -74,8 +74,8 @@ RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
         libcgal-dev \
     && apt-get clean && rm -rf /var/lib/apt/lists/*
 
-# ── PyTorch (CUDA 12.1 wheel) ────────────────────────────────────────────────
-RUN python3 -m pip install -U --extra-index-url https://download.pytorch.org/whl/cu121 \
+# ── PyTorch (CUDA 12.6 wheel line) ───────────────────────────────────────────
+RUN python3 -m pip install -U --extra-index-url https://download.pytorch.org/whl/cu126 \
     torch torchvision torchaudio
 
 # ── General Python packages ──────────────────────────────────────────────────
@@ -216,4 +216,4 @@ USER ${USERNAME}
 HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
     CMD ros2 topic list > /dev/null 2>&1 || exit 1
 
-CMD ["bash"]
\ No newline at end of file
+CMD ["bash"]