Skip to content

Commit 410cc2f

Browse files
authored
[Docker] Fix ml_dtypes downgrade issue introduced by TensorFlow (#17691)
The ml_dtypes>=0.5.1 was installed in the install python packages script, but the tensorflow 2.18 installation later will uninstall the ml_dtypes and install an old ml_dtypes version 0.4 instead. (https://ci.tlcpack.ai/blue/rest/organizations/jenkins/pipelines/tvm-docker/branches/main/runs/2062/nodes/50/steps/114/log/?start=0). We can install ml_dtypes right after tensorflow.
1 parent 119d6be commit 410cc2f

File tree

3 files changed

+4
-146
lines changed

3 files changed

+4
-146
lines changed

docker/install/ubuntu_install_tensorflow.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,3 +23,7 @@ set -o pipefail
2323
pip3 install \
2424
keras==3.5.0 \
2525
tensorflow==2.18.0
26+
27+
# tensorflow 2.18.0 requires ml-dtypes<0.5.0,>=0.4.0
28+
# ml_dtypes needs to be installed after tensorflow
29+
pip3 install ml_dtypes==0.5.1 --no-deps

python/tvm/contrib/tf_op/__init__.py

Lines changed: 0 additions & 20 deletions
This file was deleted.

python/tvm/contrib/tf_op/module.py

Lines changed: 0 additions & 126 deletions
This file was deleted.

0 commit comments

Comments
 (0)