正在使用哪个版本的cuda已安装tensoflow？

Question

能否确定正在使用哪个版本的cuda已安装tensoflow？

即在pytorch中，我可以这样做：torch._C._cuda_getDriverVersion()

注意：我不是在寻找类似Which TensorFlow and CUDA version combinations are compatible?

的兼容性组合

Answer 1

似乎没有用于检查CUDA版本的API。但是，有一种黑客手段，那就是打印tensorflow内部库的共享库依赖项。

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