pip или conda

Устанавливать Tensorflow с помощью pip или conda?

Оказывается, вовсе не всё равно, использовать для установка tensorflow pip или conda. И тому есть две убедительные причины:

  • Пакет conda tensorflow, начиная с версии 1.9.0, использует библиотеку Intel Math Kernel Library for Deep Neural Networks, она же MKL-DNN (присутствует в Anaconda, начиная с версии 2.5), которая более чем в 8 раз производительнее при вычислениях на CPU, чем установка через pip. Более того, библиотека MKL ускоряет и вычисления для пакетов NumPy, NumpyExr, SciPy и Scikit-Learn.

Этого было бы более чем достаточно, но есть и вторая причина:

  • Установка с помощью conda tensorflow-gpu автоматически устанавливает библиотеки CUDA и CuDNN для GPU-вычислений. В случае pip это придется делать руками.

Об оптимизации библиотеки MKL можно прочитать здесь.

Оригинал: VK.com

Добавить комментарий

Ваш адрес email не будет опубликован. Обязательные поля помечены *