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
-
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
Uh oh!
There was an error while loading. Please reload this page.
-
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