Patch TensorFlow pipfile to download its own cuda version GPU

opendatahub-io · Oct 2, 2024 · e5bc4f3 · e5bc4f3
1 parent eb433fc
commit e5bc4f3
Show file tree

Hide file tree

Showing 4 changed files with 841 additions and 672 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"