diff --git a/CMakeLists.txt b/CMakeLists.txt
index 0ee1e3bbc0..fdbc67141e 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -428,8 +428,7 @@ endif()
 
 if(ENABLE_DEEPKS)
   # Torch uses outdated components to detech CUDA arch, causing failure on latest CUDA kits.
-  # See above for setting CMAKE_CUDA_ARCHITECTURES
-  set(TORCH_CUDA_ARCH_LIST CMAKE_CUDA_ARCHITECTURES)
+  # Set CMake variable TORCH_CUDA_ARCH_LIST in the form of "major.minor" if required.
   find_package(Torch REQUIRED)
   if(NOT Torch_VERSION VERSION_LESS "2.1.0")
     set_if_higher(CMAKE_CXX_STANDARD 17)