[skip ci] Update script prepare_gpu_build.sh for nightly build (#648)

Signed-off-by: Cai Yudong <[email protected]>
zilliztech · Jun 14, 2024 · 6ee20d3 · 6ee20d3
1 parent 8ad9fca
commit 6ee20d3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/prepare_gpu_build.sh b/scripts/prepare_gpu_build.sh
@@ -3,5 +3,6 @@
 # This script is to modify CMakeLists.txt to build knowhere for certain GPU serials
 # GTX 1060 - sm_61
 # GTX 1660 - sm_75
-sed 's/set(CMAKE_CUDA_ARCHITECTURES .*$/set(CMAKE_CUDA_ARCHITECTURES \"61-real;75-real\")/' CMakeLists.txt > tmp
+# GTX 2080 SUPER - sm_75
+sed 's/set(CMAKE_CUDA_ARCHITECTURES .*$/set(CMAKE_CUDA_ARCHITECTURES \"75-real\")/' CMakeLists.txt > tmp
 mv tmp CMakeLists.txt