Talaan ng mga Nilalaman:

Paano ko makukuha ang TensorFlow na gamitin ang aking GPU?
Paano ko makukuha ang TensorFlow na gamitin ang aking GPU?

Video: Paano ko makukuha ang TensorFlow na gamitin ang aking GPU?

Video: Paano ko makukuha ang TensorFlow na gamitin ang aking GPU?
Video: MEGA Chia GPU Farming and Plotting Guide for Linux - Gigahorse Start to Finish - 2023 2024, Nobyembre
Anonim

Mga hakbang:

  1. I-uninstall iyong luma tensorflow .
  2. I-install tensorflow - gpu pag-install ng pip tensorflow - gpu .
  3. I-install ang Nvidia Graphics Card & Mga Driver (malamang mayroon ka na)
  4. I-download at I-install ang CUDA.
  5. I-download at I-install ang cuDNN.
  6. I-verify sa pamamagitan ng simpleng programa.

Higit pa rito, kailangan ba ng TensorFlow ng GPU?

TensorFlow GPU suporta nangangailangan isang assortment ng mga driver at library. Itong setup lang nangangailangan ang NVIDIA® GPU mga driver. Ang mga tagubilin sa pag-install na ito ay para sa pinakabagong release ng TensorFlow . Tingnan ang nasubok na mga configuration ng build para sa mga bersyon ng CUDA at cuDNN na gagamitin sa mas luma TensorFlow naglalabas.

Gayundin, paano ako gagamit ng maraming GPU sa TensorFlow? Kung ang TensorFlow Ang operasyon ay may parehong CPU at GPU mga pagpapatupad, TensorFlow ay awtomatikong ilalagay ang operasyon upang tumakbo sa a GPU device muna. Kung mayroon kang higit sa isa GPU , ang GPU na may pinakamababang ID ay pipiliin bilang default. gayunpaman, TensorFlow hindi naglalagay ng mga operasyon sa maraming GPU awtomatiko.

Habang pinapanatili itong nakikita, ano ang GPU sa TensorFlow?

A GPU (Graphical Processing Unit) ay isang bahagi ng karamihan sa mga modernong computer na idinisenyo upang magsagawa ng mga computations na kailangan para sa 3D graphics.

Maaari bang tumakbo ang TensorFlow sa AMD GPU?

Ang code na ito maaaring tumakbo katutubong sa AMD pati na rin ang Nvidia GPU . Oo posible na magpatakbo ng tensorflow sa Mga AMD GPU ngunit ito ay magiging isang ano ba ng isang problema. Bilang tensorflow gumagamit ng CUDA na pagmamay-ari nito pwede 't tumakbo sa AMD GPU's kaya kailangan mong gumamit ng OPENCL para doon at tensorflow ay hindi nakasulat doon.

Inirerekumendang: