instalar como tensorflow cuda version compatibility cudnn

como - ¿Qué combinaciones de versiones de TensorFlow y CUDA son compatibles?



install tensorflow windows (4)

Generalmente:

Compruebe la versión de CUDA:

cat /usr/local/cuda/version.txt

y versión cuDNN:

grep CUDNN_MAJOR -A 2 /usr/local/cuda/include/cudnn.h

e instale una combinación como se indica a continuación en las imágenes o here .

Las siguientes imágenes y el enlace proporcionan una descripción general de las combinaciones oficialmente admitidas / probadas de CUDA y TensorFlow en Linux, macOS y Windows:

Configuraciones menores:

Debido a que las especificaciones dadas a continuación en algunos casos pueden ser demasiado amplias, aquí hay una configuración específica que funciona:

  • tensorflow-gpu==1.12.0
  • cuda==9.0
  • cuDNN==7.1.4

El cudnn correspondiente se puede descargar here .

(Cifras actualizadas el 29 de junio de 2019).

GPU de Linux

Linux

GPU macOS

Mac OS

(figura actualizada al 31 de mayo de 2018).

Windows

Me he dado cuenta de que algunas versiones más recientes de TensorFlow son incompatibles con versiones anteriores de CUDA y cuDNN. ¿Existe una descripción general de las versiones compatibles o incluso una lista de combinaciones oficialmente probadas? No puedo encontrarlo en la documentación de TensorFlow.

La tabla de compatibilidad dada en here no contiene versiones menores específicas para cuda y cuDNN. Generalmente solo se enumera como cuda = 9 y cuDNN = 7. Sin embargo, si las versiones específicas no se cumplen, habrá un error.

Para tensorflow-gpu==1.12.0 y cuda==9.0 , la versión cuDNN compatible es 7.1.4 , que puede descargarse desde here después del registro.

Puedes consultar tu versión de cuda usando
nvcc --version

versión cuDNN usando
cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2

versión tensorflow-gpu usando
pip freeze | grep tensorflow-gpu

Puede usar esta configuración para cuda 10.0 (10.1 no funciona a partir de 3/18), esto se ejecuta para mí:

  • tensorflow> = 1.12.0
  • tensorflow_gpu> = 1.4

Instalar la versión tensorflow gpu:

pip install tensorflow-gpu==1.4.0

Trabajo: tensorflow 1.13.1, CUDA 10, CUDNN 7.4.2, python 3.6 (no funciona bien con 3.7 .. 3.7 tiene muchos errores) Para Windows 10