build axlearn with tensorflow-cpu

Steboss · Steboss · commit 0de66b0d2cf2 · 2025-02-20T17:45:44.000Z
diff --git a/.github/container/Dockerfile.axlearn b/.github/container/Dockerfile.axlearn
@@ -24,6 +24,7 @@ portpicker==1.6.0
 seqio==0.0.18
 protobuf==3.20.3  
 pytest>=7.4.3
+tensorflow-cpu
 REQUIREMENTS
 EOF
 
