From 39da4034f56f9ce9e1499f024d301c8d39300413 Mon Sep 17 00:00:00 2001
From: Manish Khadka <61139563+pen-pal@users.noreply.github.com>
Date: Tue, 2 Sep 2025 10:56:34 +0545
Subject: [PATCH] update docker network

allow option to pass docker network name or run in bridge by default
---
 crates/cuda/src/lib.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/crates/cuda/src/lib.rs b/crates/cuda/src/lib.rs
index 8bba067ef2..f99c81986a 100644
--- a/crates/cuda/src/lib.rs
+++ b/crates/cuda/src/lib.rs
@@ -205,6 +205,7 @@ impl SP1CudaProver {
         let container_name = port.map(|p| format!("sp1-gpu-{p}")).unwrap_or("sp1-gpu".to_string());
         let image_name = std::env::var("SP1_GPU_IMAGE")
             .unwrap_or_else(|_| "public.ecr.aws/succinct-labs/moongate:v5.0.8".to_string());
+         let network = std::env::var("SP1_DOCKER_NETWORK").unwrap_or("bridge".to_string());
 
         let cleaned_up = Arc::new(AtomicBool::new(false));
         let port = port.unwrap_or(3000);
@@ -232,6 +233,8 @@ impl SP1CudaProver {
                 "--rm",
                 "--gpus",
                 &gpus,
+                "--network",
+                &network,
                 "--name",
                 &container_name,
                 &image_name,