Turinys:

Kaip priversti TensorFlow naudoti mano GPU?
Kaip priversti TensorFlow naudoti mano GPU?

Video: Kaip priversti TensorFlow naudoti mano GPU?

Video: Kaip priversti TensorFlow naudoti mano GPU?
Video: MJC Stream: Видишь енота? А он есть! Главное об ML и компьютерном зрении 2024, Gegužė
Anonim

Veiksmai:

  1. Pašalinkite tavo senas tensorflow .
  2. Diegti tensorflow - gpu pip diegimas tensorflow - gpu .
  3. Įdiekite Nvidia Vaizdo plokštė & Tvarkyklės (tikriausiai jau turite)
  4. Atsisiųskite ir įdiekite CUDA.
  5. Atsisiųskite ir įdiekite cuDNN.
  6. Patikrinkite paprasta programa.

Be to, ar TensorFlow reikia GPU?

TensorFlow GPU parama reikalauja vairuotojų ir bibliotekų asortimentas. Tik ši sąranka reikalauja NVIDIA® GPU vairuotojai. Šios diegimo instrukcijos skirtos naujausiai TensorFlow . Žr. išbandytas CUDA ir cuDNN versijų kūrimo konfigūracijas, skirtas naudoti su senesnėmis versijomis TensorFlow išleidžia.

Taip pat kaip „TensorFlow“naudoti kelis GPU? Jeigu TensorFlow operacija turi ir CPU, ir GPU įgyvendinimai, TensorFlow automatiškai pradės vykdyti operaciją įjungta a GPU pirmiausia prietaisas. Jei turite daugiau nei vieną GPU , GPU pagal numatytuosius nustatymus bus pasirinktas su mažiausiu ID. Tačiau TensorFlow neatlieka operacijų keli GPU automatiškai.

Turint tai omenyje, kas yra „TensorFlow“GPU?

A GPU (Graphical Processing Unit) yra daugelio šiuolaikinių kompiuterių komponentas, skirtas atlikti 3D grafikai reikalingus skaičiavimus.

Ar „TensorFlow“gali veikti naudojant AMD GPU?

Šis kodas gali bėgti gimtoji AMD taip pat Nvidia GPU . Taip galima paleisti tensorflow įjungta AMD GPU bet tai būtų didelė problema. Kaip tensorflow naudoja CUDA, kuri yra patentuota gali ne veikia AMD GPU taigi tam reikia naudoti OPENCL ir tensorflow tame neparašyta.

Rekomenduojamas: