TensorFlow: switch dsoloader path for libcuda to end in .so.1

Vijay Vasudevan · Vijay Vasudevan · commit 431ad4be4171 · 2016-02-19T09:08:24.000-08:00
Change: 114990321
diff --git a/tensorflow/stream_executor/dso_loader.cc b/tensorflow/stream_executor/dso_loader.cc
@@ -69,9 +69,9 @@ string GetCudnnVersion() { return ""; }
 }
 
 /* static */ port::Status DsoLoader::GetLibcudaDsoHandle(void** dso_handle) {
-  return GetDsoHandle(FindDsoPath("libcuda.so",
-                                  "third_party/gpus/cuda/driver/lib64"),
-                      dso_handle);
+  return GetDsoHandle(
+      FindDsoPath("libcuda.so.1", "third_party/gpus/cuda/driver/lib64"),
+      dso_handle);
 }
 
 /* static */ port::Status DsoLoader::GetLibcuptiDsoHandle(void** dso_handle) {

Original file line number	Diff line number	Diff line change
`@@ -69,9 +69,9 @@ string GetCudnnVersion() { return ""; }`
`69`	`69`	`}`
`70`	`70`
`71`	`71`	`/* static / port::Status DsoLoader::GetLibcudaDsoHandle(void* dso_handle) {`
`72`		`- return GetDsoHandle(FindDsoPath("libcuda.so",`
`73`		`- "third_party/gpus/cuda/driver/lib64"),`
`74`		`- dso_handle);`
	`72`	`+ return GetDsoHandle(`
	`73`	`+ FindDsoPath("libcuda.so.1", "third_party/gpus/cuda/driver/lib64"),`
	`74`	`+ dso_handle);`
`75`	`75`	`}`
`76`	`76`
`77`	`77`	`/* static / port::Status DsoLoader::GetLibcuptiDsoHandle(void* dso_handle) {`