Installing torch and triton first so that gradlib is built compatible

ROCm · Nov 1, 2024 · 0f820f3 · 0f820f3
1 parent 3103abb
commit 0f820f3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -52,6 +52,11 @@ jobs:
       fail-fast: false
 
     steps:
+      - name: Prepare
+        run: |
+          pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2
+          pip3 install -U triton
+
       - name: Checkout
         uses: actions/checkout@eef61447b9ff4aafe5dcd4e0bbf5d482be7e7871 # v4.2.1