Sadržaj:

Kako mogu natjerati TensorFlow da koristi moj GPU?
Kako mogu natjerati TensorFlow da koristi moj GPU?

Video: Kako mogu natjerati TensorFlow da koristi moj GPU?

Video: Kako mogu natjerati TensorFlow da koristi moj GPU?
Video: MJC Stream: Видишь енота? А он есть! Главное об ML и компьютерном зрении 2024, Svibanj
Anonim

Koraci:

  1. Deinstaliraj tvoj star tenzorski protok .
  2. Instalirati tenzorski protok - gpu pip instalirati tenzorski protok - gpu .
  3. Instalirajte Nvidiu Grafička kartica & Driveri (vjerojatno već imate)
  4. Preuzmite i instalirajte CUDA.
  5. Preuzmite i instalirajte cuDNN.
  6. Provjerite jednostavnim programom.

Nadalje, treba li TensorFlow GPU?

TensorFlow GPU podrška zahtijeva asortiman vozača i knjižnica. Samo ovo postavljanje zahtijeva NVIDIA® GPU vozači. Ove upute za instalaciju odnose se na najnovije izdanje TensorFlow . Pogledajte testirane konfiguracije gradnje za CUDA i cuDNN verzije za korištenje sa starijim TensorFlow izdanja.

Isto tako, kako mogu koristiti više GPU-ova u TensorFlowu? Ako je a TensorFlow rad ima i CPU i GPU implementacije, TensorFlow automatski će pokrenuti operaciju na a GPU uređaj prvi. Ako imate više od jednog GPU , the GPU s najnižim ID-om bit će odabran prema zadanim postavkama. Međutim, TensorFlow ne postavlja operacije u više GPU-ova automatski.

Imajući to na umu, što je GPU u TensorFlowu?

A GPU (Grafička procesorska jedinica) je komponenta većine modernih računala koja je dizajnirana za izvođenje računanja potrebnih za 3D grafiku.

Može li TensorFlow raditi na AMD GPU-u?

Ovaj kod može trčati izvorno na AMD kao i Nvidia GPU . Da, moguće je pokrenuti tensorflow na AMD GPU-ovi ali to bi bio pravi problem. Kao tenzorski protok koristi CUDA koja je vlasnička limenka ne radi na AMD GPU-ima pa za to trebate koristiti OPENCL i tenzorski protok nije napisano u tome.

Preporučeni: