Какая установленная версия cuda, установленная tenenslow, использует? - PullRequest
Новая
       15

Какая установленная версия cuda, установленная tenenslow, использует?

0 голосов
/ 24 марта 2019

Можно ли определить, какую установленную версию cuda использует установленный tenSoflow?

т.е. в pytorch я могу сделать: torch._C._cuda_getDriverVersion()

Примечание: я не ищу комбинации совместимостинапример Какие комбинации версий TensorFlow и CUDA совместимы?

1 Ответ

1 голос
/ 25 марта 2019

Похоже, что нет API для проверки версии CUDA. Но есть хакерский способ - распечатать зависимости разделяемой библиотеки внутренней библиотеки тензорного потока. 

python -c 'import tensorflow as tf; print(tf.sysconfig.get_lib() + "/python/_pywrap_tensorflow_internal.so")' | xargs ldd |grep cuda

libcublas.so.10.0 => /usr/local/cuda/lib64/libcublas.so.10.0 (0x00007f398a94e000)
libcusolver.so.10.0 => /usr/local/cuda/lib64/libcusolver.so.10.0 (0x00007f3982267000)
libcudart.so.10.0 => /usr/local/cuda/lib64/libcudart.so.10.0 (0x00007f3981fed000)

https://github.com/tensorflow/tensorflow/issues/10827

...