Merge pull request #296 from google-deepmind:dependabot/docker/dot-devcontainer/nvidia/cuda-12.8.1-devel-ubuntu22.04

PiperOrigin-RevId: 742758540
Change-Id: Idbd6852a78933d55344aedd1b1867583df9495f7

Author: copybara-github

.devcontainer/Dockerfile

@@ -1,6 +1,6 @@
 # Use Nvidia Ubuntu 22 base (includes CUDA if a supported GPU is present)
 # https://hub.docker.com/r/nvidia/cuda
-FROM nvidia/cuda:12.8.0-devel-ubuntu22.04@sha256:54f18e2a8e1b3d03f77b9a6dc905533da46ac93a5513f10e8ba8e560db9fa5ab
+FROM nvidia/cuda:12.8.1-devel-ubuntu22.04@sha256:a99a1860ba8e2916e5c3e73b72ec4c4301653a84586e05bfc9a2aa2d58027e97
 
 ARG PYTHON_VERSION