Para ejecutar un programa de tensorflow, debes instalar la versión 1.0 de tensorflow. Luego debe instalar cuda8.0, cudnn5.0 y python3.6. Porque otras versiones siempre informan errores.
Ya instalé cuda8.0 y cudnn5.0, por lo que puede descargarlos directamente, ponerlos en un directorio y modificar .bashrc. A continuación se mencionará qué directorio colocar, cómo modificar .bashrc. La última es instalar tensorflow1.0.0. Puede que haya más pasos, pero paso a paso no supone ningún problema.
(Enlace: https://pan.baidu.com/s/1mUzm3s-6asEawtKEIjUexQ
código de extracción: 1234)
1. Primero cree un entorno virtual
conda create -n tf1.0 python=3.6
2. Ingrese a este entorno
source activate tf1.0
3. Busque la ubicación donde está instalado el entorno virtual. Mi ubicación aquí es:
4. Ingrese al directorio de este entorno virtual, que es tf.10. Este es el directorio que queremos poner.
cd tf1.0
5. Coloque la carpeta descargada cu8.0 en tf1.0
6. Modifique .bashrc, agregue las 3 líneas de código en el cuadro rojo, luego guarde y salga. No te olvides de buscar .bashrc
Compruebe si tiene éxito:
nvcc --version
7. Instale tensorflow1.0.0
pip install tensorflow-gpu==1.0.0
prueba:
Los pasos anteriores son efectivos.
ps: La versión cuda del sistema en sí es cuda9.0. Entonces, cada vez que cambio entre diferentes versiones de cuda, elimino las tres nuevas líneas en .bashrc. Se siente muy molesto. Si alguien conoce una forma más conveniente de cambiar, por favor dé algunos consejos.