pytorch
diff --git a/‎torch/CMakeLists.txt
Copy file name to clipboardExpand all lines: torch/CMakeLists.txt
+1-1Lines changed: 1 addition & 1 deletion b/‎torch/CMakeLists.txt
Copy file name to clipboardExpand all lines: torch/CMakeLists.txt
+1-1Lines changed: 1 addition & 1 deletion
@@ -481,7 +481,7 @@ add_custom_target(
   gen_torch_version ALL
   "${Python_EXECUTABLE}" "${TOOLS_PATH}/generate_torch_version.py"
     --is-debug=${TORCH_VERSION_DEBUG}
-    --cuda-version=${CUDAToolkit_VERSION}
+    --cuda-version=${CUDAToolkit_VERSION_MAJOR}.${CUDAToolkit_VERSION_MINOR}
     --hip-version=${HIP_VERSION}
     --xpu-version=${SYCL_COMPILER_VERSION}
   BYPRODUCTS ${TORCH_SRC_DIR}/version.py