update

gpu-mode · Nov 11, 2024 · 050dace · 050dace
1 parent 42d4cf5
commit 050dace
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.github/workflows/train_workflow.yml b/.github/workflows/train_workflow.yml
@@ -1,4 +1,4 @@
-name: Training Workflow
+name: AMD PyTorch Job
 on:
   workflow_dispatch:
     inputs:
@@ -13,7 +13,8 @@ jobs:
     steps:
       - name: Install dependencies
         run: |
-          pip install numpy torch
+          pip install numpy
+          pip install torch --index-url https://download.pytorch.org/whl/rocm6.2
 
       - name: Create and run training script
         run: |