experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

TensorFlow: change cuda-diagnostics to search for so.1

Browse Source 
Change: 115010103
This commit is contained in:
Vijay Vasudevan 2016-02-18 15:29:33 -08:00 committed by TensorFlower Gardener
parent 5fe957f055
commit 8174381077
1 changed files with 1 additions and 1 deletions

2
tensorflow/stream_executor/cuda/cuda_diagnostics.cc
View File

@ -165,7 +165,7 @@ port::StatusOr<DriverVersion> Diagnostician::FindDsoVersion() {
// DSO and yields its version number into the callback data, when found.
auto iterate_phdr =
[](struct dl_phdr_info *info, size_t size, void *data) -> int {
if (strstr(info->dlpi_name, "libcuda.so")) {
if (strstr(info->dlpi_name, "libcuda.so.1")) {
VLOG(1) << "found DLL info with name: " << info->dlpi_name;
char resolved_path[PATH_MAX] = {0};
if (realpath(info->dlpi_name, resolved_path) == nullptr) {