# Question about tf version for ray data and ray train

**URL:** <https://discuss.ray.io/t/question-about-tf-version-for-ray-data-and-ray-train/4335>\
**Category:** Ray Data\
**Created:** [December 3, 2021, 9:16am UTC](https://discuss.ray.io/t/question-about-tf-version-for-ray-data-and-ray-train/4335 "2021-12-03T09:16:19Z")\
**Posts on this page:** 3\
**Page:** 1

<div class="post-metadata">

**Author:** ![chuckhope](https://sea2.discourse-cdn.com/flex020/user_avatar/discuss.ray.io/chuckhope/32/1816_2.png) [@chuckhope](https://discuss.ray.io/u/chuckhope)\
**Post date:** [December 3, 2021, 9:16am UTC](https://discuss.ray.io/t/question-about-tf-version-for-ray-data-and-ray-train/4335/1 "2021-12-03T09:16:19Z")

</div>

Hi there, I wonder what version of TensorFlow is compatible with the ray train and ray data. I have tested tf 2.2 with ray data and found ray data 1.8 only support tf\>2.3 with the special argument “output\_signature” in ```  
from\_generator( generator, output\_types=None, output\_shapes=None, args=None, output\_signature=None, name=None)

```auto

Thank you~
```

---

<div class="post-metadata">

**Author:** ![Clark\_Zinzow](https://sea2.discourse-cdn.com/flex020/user_avatar/discuss.ray.io/clark_zinzow/32/445_2.png) [@Clark\_Zinzow](https://discuss.ray.io/u/Clark_Zinzow)\
**Post date:** [January 7, 2022, 12:18am UTC](https://discuss.ray.io/t/question-about-tf-version-for-ray-data-and-ray-train/4335/2 "2022-01-07T00:18:17Z")

</div>

Hi @chuckhope, yes our use of `tf.data.Dataset.from_generator()` does require `tensorflow>=2.4`. Does TensorFlow 2.4 work for you?

---

<div class="post-metadata">

**Author:** ![chuckhope](https://sea2.discourse-cdn.com/flex020/user_avatar/discuss.ray.io/chuckhope/32/1816_2.png) [@chuckhope](https://discuss.ray.io/u/chuckhope)\
**Post date:** [January 10, 2022, 1:40am UTC](https://discuss.ray.io/t/question-about-tf-version-for-ray-data-and-ray-train/4335/3 "2022-01-10T01:40:04Z")

</div>

I am afraid no. I would appreciate it if you could downgrade a version for the potential usage of the low version TensorFlow. Thank you.