Will build Windows with GPU support

chidiwilliams · Sep 30, 2024 · 26a6979 · 26a6979
1 parent 0f7d3fd
commit 26a6979
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -77,6 +77,12 @@ jobs:
           sudo apt-get install libyaml-dev libxkbcommon-x11-0 libxcb-icccm4 libxcb-image0 libxcb-keysyms1 libxcb-randr0 libxcb-render-util0 libxcb-xinerama0 libxcb-shape0 libxcb-cursor0 libportaudio2 gettext libpulse0 libgl1-mesa-dev
         if: "startsWith(matrix.os, 'ubuntu-')"
 
+      - name: Upgrade torch to GPU version for Windows
+        run: |
+          pip3 uninstall -y torch torchaudio
+          pip3 install torch==2.2.1+cu121 torchaudio==2.2.1+cu121 --index-url https://download.pytorch.org/whl/cu121
+        if: startsWith(matrix.os, 'windows-')
+
       - name: Test
         run: |
           poetry run make test
@@ -146,6 +152,12 @@ jobs:
           sudo apt-get install libyaml-dev libxkbcommon-x11-0 libxcb-icccm4 libxcb-image0 libxcb-keysyms1 libxcb-randr0 libxcb-render-util0 libxcb-xinerama0 libxcb-shape0 libxcb-cursor0 libportaudio2 gettext libpulse0 libgl1-mesa-dev
         if: "startsWith(matrix.os, 'ubuntu-')"
 
+      - name: Upgrade torch to GPU version for Windows
+        run: |
+          pip3 uninstall -y torch torchaudio
+          pip3 install torch==2.2.1+cu121 torchaudio==2.2.1+cu121 --index-url https://download.pytorch.org/whl/cu121
+        if: startsWith(matrix.os, 'windows-')
+
       - name: Install FPM
         run: gem install fpm
         if: "startsWith(matrix.os, 'ubuntu-')"