Оглавление:

Как заставить TensorFlow использовать мой графический процессор?
Как заставить TensorFlow использовать мой графический процессор?

Видео: Как заставить TensorFlow использовать мой графический процессор?

Видео: Как заставить TensorFlow использовать мой графический процессор?
Видео: Глубокое обучение с TensorFlow на GPU 2024, Май
Anonim

Шаги:

  1. Удалить ваш Старый тензорный поток .
  2. Установить тензорный поток - ГПУ pip install тензорный поток - ГПУ .
  3. Установите Nvidia Видеокарта & Драйверы (вероятно, у вас уже есть)
  4. Загрузите и установите CUDA.
  5. Загрузите и установите cuDNN.
  6. Проверяем простой программой.

Кроме того, нужен ли TensorFlow графический процессор?

Графический процессор TensorFlow служба поддержки требует набор драйверов и библиотек. Только эта установка требует NVIDIA® GPU драйверы. Эти инструкции по установке относятся к последней версии TensorFlow . См. Протестированные конфигурации сборки для версий CUDA и cuDNN для использования со старыми версиями. TensorFlow выпускает.

Аналогичным образом, как мне использовать несколько графических процессоров в TensorFlow? Если TensorFlow операция имеет как ЦП, так и GPU реализации, TensorFlow автоматически поместит операцию для запуска на а GPU устройство в первую очередь. Если у вас более одного GPU , то GPU с наименьшим идентификатором будет выбран по умолчанию. Тем не мение, TensorFlow не помещает операции в несколько графических процессоров автоматически.

Имея это в виду, что такое графический процессор в TensorFlow?

А GPU (Графический процессор) - это компонент большинства современных компьютеров, предназначенный для выполнения вычислений, необходимых для трехмерной графики.

Может ли TensorFlow работать на графическом процессоре AMD?

Этот код могу бегать изначально на AMD а также Nvidia GPU . Да это возможно запустить тензорный поток на AMD GPU но это было бы большой проблемой. В качестве тензорный поток использует CUDA, который является проприетарным жестяная банка т работать на AMD GPU поэтому вам нужно использовать OPENCL для этого и тензорный поток в этом не написано.

Рекомендуемые: