Блог

Главная - Вопросы по программированию - Tensorflow из исходного кода работает медленнее, чем версия, установленная из pip

Tensorflow из исходного кода работает медленнее, чем версия, установленная из pip

#python #tensorflow

#python #tensorflow

Вопрос:

Когда я собираю Tensorflow 1.13 из исходного кода, я заметил, что он работает медленнее, чем tensorflow-gpu, установленный из pip

  1. Графический процессор: nvidia gtx 1080
  2. Драйвер: 410
  3. Cuda: 10.0
  4. Cudnn 7.5

Я попробовал cudnn 7.3. Я установил путь к cuda и cudnn в configure. С помощью pip у меня скорость около 0,06 секунды и 0,1 от исходного
Как я могу это исправить?

Комментарии:

1. Вы скомпилировали исходный код со всеми включенными параметрами оптимизации?

2. Я использую эту команду: bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package . Также я попробовал этот: bazel build -c opt --copt=-mavx --copt=-mavx2 --copt=-mfma --copt=-mfpmath=both --copt=-msse4.2 --config=cuda //tensorflow/tools/pip_package:build_pip_package