Update amd-health.yml

Author: Mark Saroufim

.github/workflows/amd-health.yml

@@ -21,7 +21,7 @@ jobs:
 
     - name: Install PyTorch
       run: |
-        pip install torch
+        pip install torch --index-url https://download.pytorch.org/whl/rocm6.3
     
     - name: GPU Health Check
       run: python -c "import torch; torch.randn(5, device='cuda')"