ROCm / tensorflow-upstream

TensorFlow ROCm port
https://tensorflow.org
Apache License 2.0
683 stars 93 forks source link

[DRAFT] add support of MI30X for tf1.15 + py310 [pending, requirements changed] #2551

Open yiakwy-xpu-ml-framework-team opened 1 month ago

yiakwy-xpu-ml-framework-team commented 1 month ago

Description

TF 1.15 is very old which does to consider the changes of Numpy ABI and updates after python 3.10.

So we need to fix those problems to make sure TF 1.15 is well supported for existing customers.

The difficulty of changes of Numpy since 1.19 requires us to have tensorflow build numpy 1.19 by applying changes in numpy side.

However python 3.10 does not work with numpy < 1.24, due to numpy ABI changes which requires more changes from numpy side (unlike to apply patches upon python + Cython)

Soluiton

Two relase for Tf1.15:

Both versions have fixed protobuf 3.8.0 but we need make sure protobuf works well with python3.x.

Previously we only have ROC6.1 image for python 3.10 and tensorflow 2.x, for the successive release we should have

yiakwy-xpu-ml-framework-team commented 1 month ago

thank you @fsx950223 for the fix to support py38+tf1.15, the change will be relfected on dockerfile used to build the env.