temp pip install amd torch

gpu-mode · Jan 22, 2025 · af733a7 · af733a7
1 parent c625f7d
commit af733a7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/amd_workflow.yml b/.github/workflows/amd_workflow.yml
@@ -46,7 +46,7 @@ jobs:
         python -m venv ${{ matrix.venv-dir }}
         source ${{ matrix.venv-dir }}/bin/activate
         pip install --upgrade pip
-
+        pip install torch pytorch-triton-rocm --index-url https://download.pytorch.org/whl/nightly/rocm6.2
         if [[ -n "${{ github.event.inputs.requirements }}" ]]; then
           cat > "requirements.txt" <<'EOL'
           ${{ github.event.inputs.requirements }}