fix: downgraded tensorflow to 2.15.1 in the GPU docker image as 2.16.1 can't detect a GPU (#28578)

vedpatwardhan · web-flow · commit fcefb841bcbb · 2024-03-13T12:33:16.000+05:30
diff --git a/docker/gpu_framework_directory.py b/docker/gpu_framework_directory.py
@@ -67,7 +67,7 @@ def install_pkg(path, pkg, base="fw/"):
         )
     elif pkg.split("==")[0] if "==" in pkg else pkg == "tensorflow":
         subprocess.run(
-            f"yes |pip install tensorflow[and-cuda] --target {path}",
+            f"yes |pip install tensorflow[and-cuda]==2.15.1 --target {path}",
             shell=True,
         )
     else:

Original file line number	Diff line number	Diff line change
`@@ -67,7 +67,7 @@ def install_pkg(path, pkg, base="fw/"):`
`67`	`67`	`)`
`68`	`68`	`elif pkg.split("==")[0] if "==" in pkg else pkg == "tensorflow":`
`69`	`69`	`subprocess.run(`
`70`		`- f"yes \|pip install tensorflow[and-cuda] --target {path}",`
	`70`	`+ f"yes \|pip install tensorflow[and-cuda]==2.15.1 --target {path}",`
`71`	`71`	`shell=True,`
`72`	`72`	`)`
`73`	`73`	`else:`