Tabla de contenido:

¿Cómo consigo que TensorFlow use mi GPU?
¿Cómo consigo que TensorFlow use mi GPU?

Video: ¿Cómo consigo que TensorFlow use mi GPU?

Video: ¿Cómo consigo que TensorFlow use mi GPU?
Video: Installing Latest TensorFlow version with CUDA, cudNN and GPU support - Step by step tutorial 2021 2024, Mayo
Anonim

Pasos:

  1. Desinstalar tu viejo tensorflow .
  2. Instalar en pc tensorflow - gpu instalación de pip tensorflow - gpu .
  3. Instalar Nvidia Tarjeta grafica & Drivers (probablemente ya los tenga)
  4. Descargue e instale CUDA.
  5. Descargue e instale cuDNN.
  6. Verifique con un programa simple.

Además, ¿TensorFlow necesita GPU?

GPU TensorFlow apoyo requiere una variedad de controladores y bibliotecas. Esta configuración solamente requiere la NVIDIA® GPU conductores. Estas instrucciones de instalación son para la última versión de TensorFlow . Vea las configuraciones de compilación probadas para las versiones CUDA y cuDNN para usar con versiones anteriores. TensorFlow lanzamientos.

Del mismo modo, ¿cómo uso varias GPU en TensorFlow? Si un TensorFlow La operación tiene tanto CPU como GPU implementaciones, TensorFlow colocará automáticamente la operación para que se ejecute sobre a GPU dispositivo primero. Si tienes mas de uno GPU , los GPU con la ID más baja se seleccionará de forma predeterminada. Sin embargo, TensorFlow no coloca operaciones en múltiples GPU automáticamente.

Teniendo esto en cuenta, ¿qué es la GPU en TensorFlow?

A GPU (Unidad de procesamiento gráfico) es un componente de la mayoría de las computadoras modernas que está diseñado para realizar los cálculos necesarios para gráficos 3D.

¿Se puede ejecutar TensorFlow en AMD GPU?

Este codigo poder correr nativamente en AMD así como Nvidia GPU . Si, es posible ejecutar tensorflow sobre GPU de AMD pero sería un gran problema. Como tensorflow usa CUDA que es propietario pueden 't ejecutar en AMD GPU entonces necesitas usar OPENCL para eso y tensorflow no está escrito en eso.

Recomendado: