Merge pull request #722 from atheo89/RHOAIENG-13129-patch

Patch TensorFlow Pipfile to download its own cuda version GPU
opendatahub-io · Oct 4, 2024 · e7c515e · e7c515e
2 parents 609ffb8 + e152a9e
commit e7c515e
Show file tree

Hide file tree

Showing 4 changed files with 400 additions and 220 deletions.
diff --git a/jupyter/tensorflow/ubi9-python-3.11/Pipfile b/jupyter/tensorflow/ubi9-python-3.11/Pipfile
@@ -7,7 +7,7 @@ name = "pypi"
 
 [packages]
 # TensorFlow packages
-tensorflow = "~=2.17.0"
+tensorflow = {version = "~=2.17.0", extras = ["and-cuda"]}
 tensorboard = "~=2.17.1"
 tf2onnx = "~= 1.16.1"