Skip to content

Commit e152a9e

Browse files
committed
Patch TensorFlow pipfile to download its own cuda version GPU
1 parent 609ffb8 commit e152a9e

File tree

4 files changed

+400
-220
lines changed

4 files changed

+400
-220
lines changed

jupyter/tensorflow/ubi9-python-3.11/Pipfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ name = "pypi"
77

88
[packages]
99
# TensorFlow packages
10-
tensorflow = "~=2.17.0"
10+
tensorflow = {version = "~=2.17.0", extras = ["and-cuda"]}
1111
tensorboard = "~=2.17.1"
1212
tf2onnx = "~= 1.16.1"
1313

0 commit comments

Comments
 (0)