Para TensorFlow, me gustaría instalar cuda y CuDNN. ¿Cómo hago eso en Ubuntu 16.04?
Mejor respuesta
Paso 0:Instale cuda desde los repositorios estándar. (Consulte ¿Cómo puedo instalar CUDA en Ubuntu 16.04?)
Paso 1:Registre una cuenta de desarrollador de nvidia y descargue cudnn aquí (alrededor de 80 MB)
Paso 2:Comprueba dónde está tu instalación de cuda. Para la instalación desde el repositorio es /usr/lib/... y /usr/include . De lo contrario, será /usr/local/cuda/ o /usr/local/cuda-<version> . Puedes comprobarlo con which nvcc o ldconfig -p | grep cuda
Paso 3:Copie los archivos:
Instalación del repositorio:
$ cd folder/extracted/contents
$ sudo cp -P include/cudnn.h /usr/include
$ sudo cp -P lib64/libcudnn* /usr/lib/x86_64-linux-gnu/
$ sudo chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn*
Instalación del archivo de ejecución:
$ cd folder/extracted/contents
$ sudo cp include/cudnn.h /usr/local/cuda/include
$ sudo cp lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*