Предкомпилированный Tensorflow - Программный способ получить версию CUDA & CUDNN, против которой она была построена
Мне интересно, есть ли программный способ выяснить, против какой версии CUDA и CUDNN была установлена установленная версия tenorflow? Например, скомпилированная установка Tensorflow может возвратить, какой CXX11_ABI_FLAG использовался во время сборки:python -c "import tensorflow as tf; print(tf.CXX11_ABI_FLAG)" -> 0
Фон следующий: я создаю операционные системы Tensorflow в соответствии с добавлением операционной системы с бинарной установкой тензорного потока. При этом используется предварительно скомпилированный TF для извлечения необходимых путей включения и флагов компиляции, чтобы убедиться, что Tensorflow и новый Op совместимы. Но поскольку наши системы имеют несколько версий CUDA и CUDNN, мне также необходимо указать путь к нужным версиям для компилятора. Например, для Cuda 8.0 следующий флаг -L /usr/local/cuda-8.0/lib64/ указать его путь. Но это также открывает путь для ошибок, поскольку операционная система теперь может быть успешно построена с другой версией CUDA/CUDNN, что позже приводит к ошибкам во время выполнения. Поэтому я хочу создать проверку безопасности, чтобы гарантировать, что пути CUDA/CUDNN ведут к тем же версиям, что и те, против которых был построен Tensorflow.