From 3ea47178f92e5eff982ff181e476a1347b56a793 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Mon, 6 Nov 2023 10:58:08 -0800 Subject: [PATCH] Switch torch latest CPU to latest torch --- .github/workflows/nv-torch-latest-cpu.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/nv-torch-latest-cpu.yml b/.github/workflows/nv-torch-latest-cpu.yml index 3c2b7301acf0..d1a221084885 100644 --- a/.github/workflows/nv-torch-latest-cpu.yml +++ b/.github/workflows/nv-torch-latest-cpu.yml @@ -26,7 +26,7 @@ jobs: - name: Install pytorch run: | - pip install torch==1.12.0+cpu torchvision==0.13.0+cpu torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cpu + pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu python -c "import torch; print('torch:', torch.__version__, torch)" python -c "import torch; print('CUDA available:', torch.cuda.is_available())"