Съдържание:

Как да накарам TensorFlow да използва моя графичен процесор?
Как да накарам TensorFlow да използва моя графичен процесор?

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

Видео: Как да накарам TensorFlow да използва моя графичен процесор?
Видео: Машинное обучение с Python! Обучение, тестирование, разделение для оценки моделей 2024, Може
Anonim

Стъпки:

  1. Деинсталиране Вашият стар тензорен поток .
  2. Инсталирай тензорен поток - gpu pip инсталиране тензорен поток - gpu .
  3. Инсталирайте Nvidia Графична карта & Драйвери (вероятно вече имате)
  4. Изтеглете и инсталирайте CUDA.
  5. Изтеглете и инсталирайте cuDNN.
  6. Проверете с проста програма.

Освен това TensorFlow има ли нужда от графичен процесор?

Графичен процесор TensorFlow поддържа изисква асортимент от драйвери и библиотеки. Само тази настройка изисква NVIDIA® графичен процесор шофьори. Тези инструкции за инсталиране са за най-новата версия на TensorFlow . Вижте тестваните конфигурации за компилация за версии на CUDA и cuDNN, които да използвате с по-стари TensorFlow издания.

По същия начин как да използвам няколко графични процесора в TensorFlow? Ако TensorFlow операция има както CPU, така и графичен процесор реализации, TensorFlow автоматично ще постави операцията за изпълнение На а графичен процесор устройство първо. Ако имате повече от един графичен процесор , на графичен процесор с най-ниския идентификатор ще бъде избран по подразбиране. Въпреки това, TensorFlow не поставя операции в множество графични процесори автоматично.

Имайки предвид това, какво е GPU в TensorFlow?

А графичен процесор (Graphical Processing Unit) е компонент на повечето съвременни компютри, който е проектиран да извършва изчисления, необходими за 3D графика.

Може ли TensorFlow да работи на AMD GPU?

Този код може да бяга нативно на AMD както и Nvidia графичен процесор . Да, възможно е стартирайте tensorflow На AMD графични процесори но би било адски проблем. Като тензорен поток използва CUDA, която е собствена мога 'T работи на AMD GPU така че трябва да използвате OPENCL за това и тензорен поток не е написано в това.

Препоръчано: