Inhaltsverzeichnis:

Wie bringe ich TensorFlow dazu, meine GPU zu verwenden?
Wie bringe ich TensorFlow dazu, meine GPU zu verwenden?

Video: Wie bringe ich TensorFlow dazu, meine GPU zu verwenden?

Video: Wie bringe ich TensorFlow dazu, meine GPU zu verwenden?
Video: Tensorflow 2 Object Detection API Tutorial with a GPU in Windows 10 2024, Kann
Anonim

Schritte:

  1. Deinstallieren Ihre alt Tensorfluss .
  2. Installieren Tensorfluss - gpu pip installieren Tensorfluss - gpu .
  3. Nvidia installieren Grafikkarte & Treiber (haben Sie wahrscheinlich schon)
  4. CUDA herunterladen und installieren.
  5. Laden Sie cuDNN herunter und installieren Sie es.
  6. Überprüfen Sie durch einfaches Programm.

Benötigt TensorFlow außerdem eine GPU?

TensorFlow-GPU Unterstützung erfordert eine Auswahl an Treibern und Bibliotheken. Nur dieses Setup erfordert die NVIDIA® GPU Fahrer. Diese Installationsanweisungen gelten für die neueste Version von TensorFlow . Sehen Sie sich die getesteten Build-Konfigurationen für CUDA- und cuDNN-Versionen zur Verwendung mit älteren an TensorFlow gibt frei.

Wie verwende ich auch mehrere GPUs in TensorFlow? Wenn eine TensorFlow Betrieb hat sowohl CPU als auch GPU Implementierungen, TensorFlow wird den Vorgang automatisch ausführen An ein GPU Gerät zuerst. Wenn Sie mehr als einen haben GPU , das GPU mit der niedrigsten ID wird standardmäßig ausgewählt. Jedoch, TensorFlow platziert keine Operationen in mehrere GPUs automatisch.

Was ist GPU in TensorFlow?

EIN GPU (Graphical Processing Unit) ist eine Komponente der meisten modernen Computer, die für die Durchführung von Berechnungen für 3D-Grafiken entwickelt wurde.

Kann TensorFlow auf einer AMD-GPU ausgeführt werden?

Dieser Code kann laufen nativ auf AMD sowie Nvidia GPU . Ja, das ist möglich Tensorflow ausführen An AMD-GPUs aber es wäre ein Problem. Wie Tensorfluss verwendet CUDA, das proprietär ist kann 'T läuft auf AMD-GPUs Sie müssen dafür OPENCL verwenden und Tensorfluss steht darin nicht geschrieben.

Empfohlen: