Bump the version of the tensorflow image in our tensorflow example (#973

)
modal-labs · Nov 14, 2024 · 8110625 · 8110625
1 parent 2a142ac
commit 8110625
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/06_gpu_and_ml/tensorflow/tensorflow_tutorial.py b/06_gpu_and_ml/tensorflow/tensorflow_tutorial.py
@@ -28,8 +28,8 @@
 import modal
 
 dockerhub_image = modal.Image.from_registry(
-    "tensorflow/tensorflow:2.12.0-gpu",
-).pip_install("protobuf==3.20.*")
+    "tensorflow/tensorflow:2.15.0-gpu",
+)
 
 app = modal.App("example-tensorflow-tutorial", image=dockerhub_image)