diff --git a/.github/actions/install/action.yml b/.github/actions/install/action.yml
index 42adae9f7..6b94ed0f2 100644
--- a/.github/actions/install/action.yml
+++ b/.github/actions/install/action.yml
@@ -53,6 +53,7 @@ runs:
         if [[ "${{ inputs.use_vm }}" == "true" ]]; then source ~/venv/bin/activate; fi
         pip install --no-cache-dir --upgrade 'pip>=20'
         pip install --no-cache-dir --upgrade 'setuptools>=68.2.2'
+        rm -rf ~/.cache/pip
 
     - name: Print pip and setuptools versions
       shell: bash
@@ -68,5 +69,5 @@ runs:
         echo "Installing graphnet with flags: ${{ steps.flags.outputs.user_flag }} ${{ steps.flags.outputs.editable_flag }}"
         echo "pip install ${{ steps.flags.outputs.user_flag }} ${{ steps.flags.outputs.editable_flag }} .[torch-${{ steps.flags.outputs.torch_flag }},${{ inputs.extras }}] -f https://data.pyg.org/whl/torch-${{ inputs.torch_version }}+${{ inputs.hardware }}.html"
         pip install --no-cache-dir${{ steps.flags.outputs.user_flag }} ${{ steps.flags.outputs.editable_flag }} .[torch-${{ steps.flags.outputs.torch_flag }},${{ inputs.extras }}] -f https://data.pyg.org/whl/torch-${{ inputs.torch_version }}+${{ inputs.hardware }}.html
-
+        rm -rf ~/.cache/pip
         pip install --no-cache-dir git+https://github.com/thoglu/jammy_flows.git
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 95555e523..242e71bc2 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -87,7 +87,8 @@ jobs:
           pip --version
           python --version
           pip show setuptools
-
+          rm -rf ~/.cache/pip
+      
       - name: Print available disk space before graphnet install
         run: | 
           df -h
@@ -102,6 +103,13 @@ jobs:
           pip install --no-cache-dir --upgrade versioneer
           pip show setuptools
           pip show versioneer
+          rm -rf ~/.cache/pip
+      - name: Install PyTorch and dependencies
+        shell: bash
+        run: |
+          source ~/venv/bin/activate
+          pip3 install --no-cache-dir torch==${{ matrix.torch_version }}+${{ matrix.hardware }} torchvision torchaudio --index-url https://download.pytorch.org/whl/${{ matrix.hardware }}
+          rm -rf ~/.cache/pip
       - name: Install package
         uses: ./.github/actions/install
         with: