diff --git a/CMakeLists.txt b/CMakeLists.txt
index f0ca595..625f062 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -119,7 +119,8 @@ if(GPU_RUNTIME STREQUAL "CUDA")
         endif()
         # Set torch cuda architecture list
         set(TORCH_CUDA_ARCH_LIST ${CMAKE_CUDA_ARCHITECTURES})
-        list(TRANSFORM TORCH_CUDA_ARCH_LIST REPLACE "([0-9])([0-9])" "\\1.\\2")
+        list(TRANSFORM TORCH_CUDA_ARCH_LIST REPLACE "^([0-9])([0-9])$" "\\1.\\2")
+        list(TRANSFORM TORCH_CUDA_ARCH_LIST REPLACE "^([0-9][0-9])([0-9])$" "\\1.\\2")
         string(REPLACE ";" " " TORCH_CUDA_ARCH_LIST "${TORCH_CUDA_ARCH_LIST}")
         message(STATUS "** Updated TORCH_CUDA_ARCH_LIST to ${TORCH_CUDA_ARCH_LIST} **")
     endif()