Taula de continguts:

Com puc fer que TensorFlow utilitzi la meva GPU?
Com puc fer que TensorFlow utilitzi la meva GPU?

Vídeo: Com puc fer que TensorFlow utilitzi la meva GPU?

Vídeo: Com puc fer que TensorFlow utilitzi la meva GPU?
Vídeo: How to Inject Your Trained Subject e.g. Your Face Into Any Custom Stable Diffusion Model By Web UI 2024, Maig
Anonim

Passos:

  1. Desinstal·la teu vell tensorflow .
  2. Instal·lar tensorflow - gpu instal·lació de pip tensorflow - gpu .
  3. Instal·leu Nvidia Targeta gràfica & Conductors (probablement ja en teniu)
  4. Descarrega i instal·la CUDA.
  5. Baixeu i instal·leu cuDNN.
  6. Verificar mitjançant un programa senzill.

A més, TensorFlow necessita GPU?

GPU TensorFlow suport requereix una varietat de controladors i biblioteques. Només aquesta configuració requereix la NVIDIA® GPU conductors. Aquestes instruccions d'instal·lació són per a la darrera versió de TensorFlow . Consulteu les configuracions de compilació provades per a les versions CUDA i cuDNN per utilitzar-les amb anteriors TensorFlow llançaments.

De la mateixa manera, com puc utilitzar diverses GPU a TensorFlow? Si a TensorFlow L'operació té tant CPU com GPU implementacions, TensorFlow posarà automàticament l'operació a executar activat a GPU primer dispositiu. Si en tens més d'un GPU , el GPU amb l'identificador més baix es seleccionarà per defecte. Malgrat això, TensorFlow no realitza operacions múltiples GPU automàticament.

Tenint això a la vista, què és la GPU a TensorFlow?

A GPU (Unitat de processament gràfic) és un component de la majoria dels ordinadors moderns dissenyat per realitzar els càlculs necessaris per als gràfics 3D.

Es pot executar TensorFlow a la GPU AMD?

Aquest codi pot córrer de manera nativa AMD així com Nvidia GPU . Sí que és possible executar tensorflow activat GPU AMD però seria un gran problema. Com tensorflow utilitza CUDA que és propietari llauna 't executar-se amb GPU AMD així que heu d'utilitzar OPENCL per a això i tensorflow no està escrit en això.

Recomanat: