diff --git a/.github/workflows/train.yml b/.github/workflows/train.yml
index b40d4cb..d536418 100644
--- a/.github/workflows/train.yml
+++ b/.github/workflows/train.yml
@@ -39,7 +39,7 @@ jobs:
           sudo apt install -y libopenmpi-dev openmpi-bin
           sudo apt-get install -y libgl1 libglib2.0-0
 
-          pip install -r requirements.txt
+          pip install -r requirements_cpu.txt
 
       # Step 4: Run gRPC server and client
       - name: Run test