does ColossalAI support tensorflow? #3510
Unanswered
robin-ai-ml
asked this question in
Community | Q&A
Replies: 2 comments
-
As far as I know, only run with Pytorch now. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I have a tensorflow keras project, now I want to integrate tensorflow project to colossalai framework.
but i read the document
Requirements:
PyTorch >= 1.11 (PyTorch 2.x in progress)
Python >= 3.7
CUDA >= 11.0
so I wonder if Colossal AI framework support tensorflow project?
or it is only run with Pytorch?
Beta Was this translation helpful? Give feedback.
All reactions