Tartalomjegyzék:

Hogyan vehetem rá a TensorFlow-t a GPU használatára?
Hogyan vehetem rá a TensorFlow-t a GPU használatára?

Videó: Hogyan vehetem rá a TensorFlow-t a GPU használatára?

Videó: Hogyan vehetem rá a TensorFlow-t a GPU használatára?
Videó: hadoop yarn architecture 2024, Lehet
Anonim

Lépések:

  1. Eltávolítás a ti régi tenzoráramlás .
  2. Telepítés tenzoráramlás - gpu pip telepítés tenzoráramlás - gpu .
  3. Telepítse az Nvidiát Grafikus kártya & Illesztőprogramok (valószínűleg már vannak)
  4. Töltse le és telepítse a CUDA-t.
  5. A cuDNN letöltése és telepítése.
  6. Ellenőrizze egyszerű programmal.

Ezenkívül a TensorFlow-nak szüksége van GPU-ra?

TensorFlow GPU támogatás igényel járművezetők és könyvtárak választéka. Csak ez a beállítás igényel az NVIDIA® GPU járművezetők. Ezek a telepítési utasítások a legújabb kiadásra vonatkoznak TensorFlow . Tekintse meg a tesztelt összeállítási konfigurációkat a CUDA és cuDNN verziókhoz, amelyeket régebbi verziókhoz használhat TensorFlow kiadja.

Hasonlóképpen, hogyan használhatok több GPU-t a TensorFlow-ban? Ha egy TensorFlow művelet rendelkezik CPU és GPU megvalósítások, TensorFlow automatikusan futtatásra helyezi a műveletet tovább a GPU először a készüléket. Ha több is van GPU , az GPU a legalacsonyabb azonosítójú lesz alapértelmezés szerint kiválasztva. Azonban, TensorFlow nem helyezi el a műveleteket több GPU automatikusan.

Ezt szem előtt tartva, mi az a GPU a TensorFlow-ban?

A GPU (Graphical Processing Unit) a legtöbb modern számítógép egyik összetevője, amelyet a 3D grafikához szükséges számítások elvégzésére terveztek.

A TensorFlow futhat AMD GPU-n?

Ezt a kódot tud futni natívan tovább AMD valamint az Nvidia GPU . Igen lehetséges futtassa a tensorflow-t tovább AMD GPU-k de nagy baj lenne. Mint tenzoráramlás CUDA-t használ, amely szabadalmaztatott tud 't AMD GPU-n fut tehát ehhez az OPENCL-t kell használnia és tenzoráramlás nincs beleírva.

Ajánlott: