Sisällysluettelo:

Kuinka saan TensorFlow'n käyttämään GPU:ta?
Kuinka saan TensorFlow'n käyttämään GPU:ta?

Video: Kuinka saan TensorFlow'n käyttämään GPU:ta?

Video: Kuinka saan TensorFlow'n käyttämään GPU:ta?
Video: Creating a Virtual Environment within a Jupyter Notebook! 2024, Marraskuu
Anonim

Askeleet:

  1. Poista asennus sinun vanha tensorflow .
  2. Asentaa tensorflow - gpu pip asennus tensorflow - gpu .
  3. Asenna Nvidia Näytönohjain & Ajurit (luultavasti sinulla on jo)
  4. Lataa ja asenna CUDA.
  5. Lataa ja asenna cuDNN.
  6. Tarkista yksinkertaisella ohjelmalla.

Lisäksi tarvitseeko TensorFlow GPU:ta?

TensorFlow GPU tuki vaatii valikoima kuljettajia ja kirjastoja. Vain tämä asetus vaatii NVIDIA® GPU Kuljettajat. Nämä asennusohjeet koskevat uusinta versiota TensorFlow . Katso testatut koontikokoonpanot CUDA- ja cuDNN-versioille käytettäväksi vanhempien kanssa TensorFlow julkaisut.

Samoin kuinka käytän useita GPU:ita TensorFlowissa? Jos TensorFlow toiminnassa on sekä CPU että GPU toteutukset, TensorFlow asettaa toiminnon automaattisesti suoritettavaksi päällä a GPU laite ensin. Jos sinulla on useampi kuin yksi GPU , GPU pienin tunnus valitaan oletusarvoisesti. Kuitenkin, TensorFlow ei aseta operaatioita useita GPU:ita automaattisesti.

Kun tämä pidetään mielessä, mikä on GPU TensorFlowissa?

A GPU (Graphical Processing Unit) on useimpien nykyaikaisten tietokoneiden komponentti, joka on suunniteltu suorittamaan 3D-grafiikassa tarvittavia laskelmia.

Voiko TensorFlow toimia AMD GPU:lla?

Tämä koodi osaa juosta alkuperäisesti päällä AMD sekä Nvidia GPU . Kyllä se on mahdollista ajaa tensorflowa päällä AMD GPU:t mutta se olisi suuri ongelma. Kuten tensorflow käyttää CUDAa, joka on sen oma voi 't toimii AMD GPU:lla joten sinun on käytettävä OPENCLiä tähän ja tensorflow siihen ei ole kirjoitettu.

Suositeltava: