Cache torch home path to prevent reassignement during install stage.

estrozi · May 29, 2024 · 5d5790f · 5d5790f
1 parent c818a03
commit 5d5790f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cmake/FindCondaPython.cmake b/cmake/FindCondaPython.cmake
@@ -97,7 +97,7 @@ if (NOT PYTHON_EXE_PATH STREQUAL "")
             RESULT_VARIABLE PYTHON_RESULT
     )
     if(EXISTS "${TORCH_HOME_PATH_OUTPUT}")
-      set(TORCH_HOME_PATH ${TORCH_HOME_PATH_OUTPUT})
+      set(TORCH_HOME_PATH ${TORCH_HOME_PATH_OUTPUT} CACHE PATH "Path to Torch home directory for storage of trained models." FORCE)
     else ()
       message(
               WARNING