diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 52d6c1a..5afd114 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -96,3 +96,5 @@ jobs:
           context: .
           push: false
           tags: chainlearn-api:ci
+          cache-from: type=gha
+          cache-to: type=gha,mode=max