From 164cc48288047d88c20d1c5929803353302ff54b Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Thu, 30 Jan 2025 08:27:55 +0800 Subject: [PATCH] fix building for cuda 12.6 --- scripts/github_actions/build-ubuntu-cuda.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/github_actions/build-ubuntu-cuda.sh b/scripts/github_actions/build-ubuntu-cuda.sh index 2cb6965..5d0ae06 100755 --- a/scripts/github_actions/build-ubuntu-cuda.sh +++ b/scripts/github_actions/build-ubuntu-cuda.sh @@ -87,7 +87,7 @@ echo "Installing torch" # /Python-3.8.2/py-3.8/lib/python3.8/site-packages/torch/share/cmake/Torch/TorchConfig.cmake:68 (find_package) # cmake/torch.cmake:14 (find_package) # CMakeLists.txt:62 (include) -sed -i.bak /9.0a/d /Python-*/py-3.*/lib/python3.*/site-packages/torch/share/cmake/Caffe2/Modules_CUDA_fix/upstream/FindCUDA/select_compute_arch.cmake +sed -i.bak /9.0a/d /Python-*/py-3.*/lib/python3.*/site-packages/torch/share/cmake/Caffe2/Modules_CUDA_fix/upstream/FindCUDA/select_compute_arch.cmake || true rm -rf ~/.cache/pip >/dev/null 2>&1 yum clean all >/dev/null 2>&1