Skip to content

Commit 6a52223

Browse files
juntaoclaude
andcommitted
Fix CUDA libtorch version: cu124 does not exist for 2.7.0, use cu126
Signed-off-by: Michael Yuan <michael@secondstate.io> Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent a848df0 commit 6a52223

3 files changed

Lines changed: 5 additions & 5 deletions

File tree

.github/workflows/release-rust.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ jobs:
9191
env:
9292
LIBTORCH_VERSION: "2.7.0"
9393
LIBTORCH_BYPASS_VERSION_CHECK: "1"
94-
CUDA_VERSION: "cu124"
94+
CUDA_VERSION: "cu126"
9595

9696
steps:
9797
- uses: actions/checkout@v4

rust/README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -300,9 +300,9 @@ Download and extract libtorch for your platform:
300300
curl -LO https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.7.1%2Bcpu.zip
301301
unzip libtorch-cxx11-abi-shared-with-deps-2.7.1+cpu.zip
302302

303-
# Linux x86_64 (CUDA 12.4)
304-
curl -LO https://download.pytorch.org/libtorch/cu124/libtorch-cxx11-abi-shared-with-deps-2.7.0%2Bcu124.zip
305-
unzip libtorch-cxx11-abi-shared-with-deps-2.7.0+cu124.zip
303+
# Linux x86_64 (CUDA 12.6)
304+
curl -LO https://download.pytorch.org/libtorch/cu126/libtorch-cxx11-abi-shared-with-deps-2.7.0%2Bcu126.zip
305+
unzip libtorch-cxx11-abi-shared-with-deps-2.7.0+cu126.zip
306306

307307
# Linux aarch64 (CPU)
308308
curl -LO https://github.com/second-state/libtorch-releases/releases/download/v2.7.1/libtorch-cxx11-abi-aarch64-2.7.1.tar.gz

rust/install.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ download_cuda_libtorch() {
119119
fi
120120

121121
info "Downloading CUDA libtorch (this may take a while)..."
122-
local url="https://download.pytorch.org/libtorch/cu124/libtorch-cxx11-abi-shared-with-deps-2.7.0%2Bcu124.zip"
122+
local url="https://download.pytorch.org/libtorch/cu126/libtorch-cxx11-abi-shared-with-deps-2.7.0%2Bcu126.zip"
123123
local temp_dir
124124
temp_dir=$(mktemp -d)
125125

0 commit comments

Comments
 (0)