File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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
Original file line number Diff line number Diff line change @@ -300,9 +300,9 @@ Download and extract libtorch for your platform:
300300curl -LO https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.7.1%2Bcpu.zip
301301unzip 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)
308308curl -LO https://github.com/second-state/libtorch-releases/releases/download/v2.7.1/libtorch-cxx11-abi-aarch64-2.7.1.tar.gz
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments