diff --git a/.github/workflows/rust.yml b/.github/workflows/rust.yml
index ab38dbfd..cdaea5e8 100644
--- a/.github/workflows/rust.yml
+++ b/.github/workflows/rust.yml
@@ -33,6 +33,12 @@ jobs:
           sudo apt-get update
           sudo apt install -y libegl1-mesa libgl1-mesa-dri libxcb-xfixes0-dev vulkan-sdk mesa-vulkan-drivers pkg-config libasound2-dev
 
+      - name: Setup python 
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.10.6'
+          cache: 'pip'
+      - run: pip install -r requirements.txt
       - name: Setup
         run: |
           cargo install wasm-pack
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 00000000..cbd0e90e
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,3 @@
+--extra-index-url https://download.pytorch.org/whl/cpu
+numpy==1.24.3
+torch==2.0.1