diff --git a/Dockerfile b/Dockerfile
index ff6eef9bd..30d69ad29 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,6 +1,10 @@
 # Base OS
 FROM ubuntu:24.04
 
+# Allow passing this variable in from the outside.
+ARG CUDA_PATH
+ENV PATH="$CUDA_PATH/bin:$PATH"
+
 # Update package list & install some basic tools we'll need.
 RUN apt-get update
 RUN apt-get install -y python3-dev python3-pip python3-venv make g++ wget git
diff --git a/install/tests/Dockerfile b/install/tests/Dockerfile
index f5efff8ec..2e8969da6 100644
--- a/install/tests/Dockerfile
+++ b/install/tests/Dockerfile
@@ -1,6 +1,10 @@
 # Base OS
 FROM ubuntu:24.04
 
+# Allow passing this variable in from the outside.
+ARG CUDA_PATH
+ENV PATH="$CUDA_PATH/bin:$PATH"
+
 # Update package list & install some basic tools we'll need.
 RUN apt-get update
 RUN apt-get install -y python3-dev python3-pip python3-venv make g++ wget git