Skip to content

Commit 74e032e

Browse files
Merge pull request #609 from atheo89/resolve-tensorflow-rocm
fix tensorflow rocm resolution by installing it from .whl file
2 parents ab84dfa + 3ec36c9 commit 74e032e

File tree

2 files changed

+231
-230
lines changed

2 files changed

+231
-230
lines changed

jupyter/amd/tensorflow/ubi9-python-3.9/Pipfile

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3,20 +3,13 @@ url = "https://pypi.org/simple"
33
verify_ssl = true
44
name = "pypi"
55

6-
[[source]]
7-
url = "https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1/"
8-
verify_ssl = true
9-
name = "tensorflow"
10-
116
[dev-packages]
127

138
[packages]
14-
# TensorFlow packages
15-
tensorflow = {version = "~=2.15.0", index = "tensorflow"}
16-
tensorboard = {version = "~=2.15.2", index = "tensorflow"}
9+
tensorflow_rocm = {file = "https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1/tensorflow_rocm-2.15.0-cp39-cp39-manylinux2014_x86_64.whl"}
10+
tensorboard = "~=2.15.2"
1711
tf2onnx = "~= 1.16.1"
1812

19-
2013
# Datascience and useful extensions
2114
boto3 = "~=1.34.50"
2215
kafka-python = "~=2.0.2"

0 commit comments

Comments
 (0)