Servidor GPU de la nube de Alibaba:
Para evitar problemas, la instalación automática del controlador de GPU se selecciona de forma predeterminada y la versión es 390.46 bajo 9.0.176;
PD: Esto se eligió al realizar un solo pedido para seleccionar el sistema operativo. Más tarde, cuando planeaba reinstalar el sistema, descubrí que el controlador no estaba seleccionado durante la reinstalación, así que me di por vencido y no lo reinstalé. No sé si enviaría una orden de trabajo para la reinstalación o no después de activar el sistema. Una vez que ingresa, puede ver el archivo de registro de la instalación del controlador, y eso es todo si se realizó correctamente;
actualizacion del sistema:
yum update
Actualice Python3, consulte https://cucldk.com/post/centos7-to-python3.html
https://developer.nvidia.com/cudnnDescargue cuDNN del sitio web oficial, debe registrar una cuenta para iniciar sesión, descargar la versión Descargar cuDNN v7.3.1 (28 de septiembre de 2018) , para cuDNN v7 en la biblioteca CUDA 9.0 .3.1 para Linux;
Las descargas nacionales de sitios web extranjeros pueden ser lentas, por lo que si tiene prisa, puede utilizar un VPS extranjero para transferirlas;
tar -zxvf cudnn-9.0-linux-x64-v7.3.1.20.tgz cd cuda cp include/* /usr/local/cuda-9.0/inlcude cp lib64/lib* /usr/local/cuda-9.0/lib64
Modificar variables de entorno y ejecutar
sudo vim /etc/profile
Agregue dos líneas de código debajo de la línea RUTA de exportación:
export PATH=/usr/local/cuda-9.0/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
Salir de la ejecución
source /etc/profile
pip3 install tensorflow-gpu==1.12.0
Ejecute el código de prueba para ver si se puede importar normalmente y generar texto personalizado:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
sess.run(hello)
ejecutar después
from tensorflow.python.client import device_lib print(device_lib.list_local_devices())
El éxito se produce cuando se detecta la gpu.
Editor de referencia: https://blog.csdn.net/Oh_My_Fish/article/details/78861867