Sadržaj:

Kako da natjeram TensorFlow da koristi moj GPU?
Kako da natjeram TensorFlow da koristi moj GPU?

Video: Kako da natjeram TensorFlow da koristi moj GPU?

Video: Kako da natjeram TensorFlow da koristi moj GPU?
Video: KAKO DA GA/JE NATERAS DA SE ZALJUBI U TEBE 2024, Maj
Anonim

Koraci:

  1. Deinstalirati vaša star tensorflow .
  2. Instaliraj tensorflow - gpu pip install tensorflow - gpu .
  3. Instalirajte Nvidia Graficka kartica & Driveri (vjerovatno već imate)
  4. Preuzmite i instalirajte CUDA.
  5. Preuzmite i instalirajte cuDNN.
  6. Provjeriti jednostavnim programom.

Štaviše, da li je TensorFlowu potreban GPU?

TensorFlow GPU podrška zahtijeva asortiman drajvera i biblioteka. Samo ovo podešavanje zahtijeva NVIDIA® GPU vozači. Ova uputstva za instalaciju su za najnovije izdanje TensorFlow . Pogledajte testirane konfiguracije gradnje za CUDA i cuDNN verzije za korištenje sa starijim TensorFlow izdanja.

Isto tako, kako da koristim više GPU-a u TensorFlow-u? Ako a TensorFlow rad ima i CPU i GPU implementacije, TensorFlow automatski će pokrenuti operaciju on a GPU uređaj prvo. Ako imate više od jednog GPU , the GPU sa najnižim ID-om će biti odabrano po defaultu. Kako god, TensorFlow ne postavlja operacije u više GPU-ova automatski.

Imajući ovo u vidu, šta je GPU u TensorFlowu?

A GPU (Jedinica za grafičku obradu) je komponenta većine modernih računara koja je dizajnirana za obavljanje proračuna potrebnih za 3D grafiku.

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

Ovaj kod može da trči izvorno na AMD kao i Nvidia GPU . Da, moguće je pokrenite tensorflow on AMD GPU-ovi ali to bi bio pravi problem. As tensorflow koristi CUDA koja je vlasnička mogu 't radi na AMD GPU-ima tako da morate koristiti OPENCL za to i tensorflow nije napisano u tome.

Preporučuje se: