Import of PR #20858.

Add alternative CUDA root that is used in some systems. PiperOrigin-RevId: 711428469
openxla · Jan 3, 2025 · d458d97 · d458d97
1 parent a84e3b7
commit d458d97
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/xla/tsl/platform/default/cuda_root_path.cc b/xla/tsl/platform/default/cuda_root_path.cc
@@ -58,6 +58,7 @@ std::vector<std::string> CandidateCudaRoots() {
 
   roots.push_back(TF_CUDA_TOOLKIT_PATH);
   roots.emplace_back(std::string("/usr/local/cuda"));
+  roots.emplace_back(std::string("/opt/cuda"));
 
 #if defined(PLATFORM_POSIX) && !defined(__APPLE__)
   Dl_info info;