Update ROCm verison (#4486)

* Update ROCm verison * Update .github/workflows/amd-mi200.yml --------- Co-authored-by: Michael Wyatt <[email protected]>
microsoft · Oct 10, 2023 · 427253b · 427253b
1 parent 706a725
commit 427253b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/amd-mi200.yml b/.github/workflows/amd-mi200.yml
@@ -28,7 +28,7 @@ jobs:
 
       - name: Install pytorch
         run: |
-          pip install -U --cache-dir $TORCH_CACHE torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.4.2
+          pip install -U --cache-dir $TORCH_CACHE torch torchvision --index-url https://download.pytorch.org/whl/rocm5.6
           python -c "import torch; print('torch:', torch.__version__, torch)"
           python -c "import torch; print('CUDA available:', torch.cuda.is_available())"