read cuda driver deb in 1M chunk

juju-solutions · Oct 30, 2023 · 28dc2b1 · 28dc2b1
1 parent 2f741b0
commit 28dc2b1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/reactive/docker.py b/reactive/docker.py
@@ -428,7 +428,8 @@ def install_cuda_drivers_repo(architecture, release, ubuntu):
 
     with contextlib.closing(urlopen(repository_url)) as r:
         with open(cuda_repository_package, "wb") as f:
-            f.write(r.read())
+            while chunk := r.read(1024 * 1024):  # 1M chunks
+                f.write(chunk)
 
     command = "dpkg -i {}".format(cuda_repository_package)
     check_call(split(command))