1. У меня есть видеокарта Quadro, и я хочу ее опробовать. Сначала я установил tensorflow и tensorrt с помощью списка conda.
Код:
импортировать тензорный поток как tf
печать (tf.test.is_built_with_cuda())
print(tf.config.list_physical_devices('GPU'))
Столкнулся со следующими проблемами:
W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] Предупреждение TF-TRT: не удалось найти TensorRT
Истинный
W tensorflow/core/common_runtime/gpu/gpu_device.cc:1960] Невозможно открыть некоторые библиотеки графического процессора. Если вы хотите использовать графический процессор, убедитесь, что недостающие библиотеки, упомянутые выше, установлены правильно. Следуйте руководству по адресу https://www.tensorflow.org/install/gpu, чтобы узнать, как загрузить и настроить необходимые библиотеки для вашей платформы.
Пропуск регистрации устройств с графическим процессором...
[]
2. Добавьте import tensorrt как trt, чтобы решить проблему, и tensorrt должен быть перед tensorflow.
Код:
импортировать tensorrt как trt
импортировать тензорный поток как tf
печать (tf.test.is_built_with_cuda())
print(tf.config.list_physical_devices('GPU'))
результат:
True
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]