Serveur GPU Alibaba Cloud :
Afin d'éviter les pièges, l'installation automatique du pilote GPU est sélectionnée par défaut, et la version est 390.46 sous 9.0.176 ;
ps : Ceci a été choisi lors d'une seule commande pour sélectionner le système d'exploitation. Plus tard, alors que j'envisageais de réinstaller le système, j'ai constaté que le pilote n'était pas sélectionné lors de la réinstallation, j'ai donc abandonné et je n'ai pas réinstallé. Je ne sais pas si je soumettrai ou non un bon de travail pour la réinstallation ; une fois le système activé, une fois entré, vous pourrez voir le fichier journal de l'installation du pilote, et c'est tout si elle réussit ;
mise à jour du système:
yum update
Mettez à jour Python3, reportez-vous à https://cucldk.com/post/centos7-to-python3.html
https://developer.nvidia.com/cudnnTéléchargez cuDNN depuis le site officiel, vous devez créer un compte pour vous connecter, téléchargez la version Téléchargez cuDNN v7.3.1 (28 septembre 2018) , pour cuDNN v7 sous CUDA 9.0 .3.1 Library pour Linux ;
Les téléchargements nationaux de sites Web étrangers peuvent être lents, donc si vous êtes pressé, vous pouvez utiliser un VPS étranger pour les transférer ;
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
Modifier les variables d'environnement et exécuter
sudo vim /etc/profile
Ajoutez deux lignes de code sous la ligne export PATH :
export PATH=/usr/local/cuda-9.0/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
Quitter l'exécution
source /etc/profile
pip3 install tensorflow-gpu==1.12.0
Exécutez le code de test pour voir s'il peut être importé normalement et afficher un texte personnalisé :
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
sess.run(hello)
exécuter après
from tensorflow.python.client import device_lib print(device_lib.list_local_devices())
Le succès se produit lorsque le GPU est détecté.
Éditeur de référence : https://blog.csdn.net/Oh_My_Fish/article/details/78861867