Server GPU Alibaba Cloud:
Per evitare insidie, di default è selezionata l'installazione automatica del driver GPU e la versione è 390.46 sotto 9.0.176;
ps: questo è stato scelto quando ho effettuato un singolo ordine per selezionare il sistema operativo. Successivamente, quando stavo progettando di reinstallare il sistema, ho scoperto che la reinstallazione non includeva la fase di selezione del driver, quindi ho rinunciato e non ho reinstallato. Non so se lo installerò se invio un ordine di lavoro per la reinstallazione dopo l'attivazione del sistema. Una volta entrato, puoi vedere il file di registro dell'installazione del driver, e questo è tutto se è andata a buon fine;
aggiornamento del sistema:
yum update
Aggiorna Python3, fai riferimento a https://cucldk.com/post/centos7-to-python3.html
https://developer.nvidia.com/cudnnScarica cuDNN dal sito Web ufficiale, è necessario registrare un account per accedere, scarica la versione Scarica cuDNN v7.3.1 (28 settembre 2018) , per cuDNN v7 in CUDA 9.0 .3.1 Libreria per Linux;
I download nazionali di siti Web stranieri potrebbero essere lenti, quindi se hai fretta puoi utilizzare un VPS straniero per trasferirli;
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
Modifica le variabili di ambiente ed esegui
sudo vim /etc/profile
Aggiungi due righe di codice sotto la riga PATH di esportazione:
export PATH=/usr/local/cuda-9.0/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
Esci dall'esecuzione
source /etc/profile
pip3 install tensorflow-gpu==1.12.0
Esegui il codice di prova per vedere se può essere importato normalmente e generare testo personalizzato:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
sess.run(hello)
eseguire dopo
from tensorflow.python.client import device_lib print(device_lib.list_local_devices())
L'esito positivo si verifica quando viene rilevata la GPU.
Editor di riferimento: https://blog.csdn.net/Oh_My_Fish/article/details/78861867