From 427253b94bf425395d48f494270550bac9ac880a Mon Sep 17 00:00:00 2001 From: Logan Adams <114770087+loadams@users.noreply.github.com> Date: Tue, 10 Oct 2023 10:41:28 -0700 Subject: [PATCH] Update ROCm verison (#4486) * Update ROCm verison * Update .github/workflows/amd-mi200.yml --------- Co-authored-by: Michael Wyatt --- .github/workflows/amd-mi200.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/amd-mi200.yml b/.github/workflows/amd-mi200.yml index a275225cc5e4..77f33f744ea8 100644 --- 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())"