web-dev-qa-db-fra.com

Comment installer Tensorflow pour NVIDIA GPU RTX 2070 SUPER sur Ubuntu 18.04

J'ai essayé d'installer Tensorflow pour mon nouveau RTX 2070 Super. J'ai suivi les instructions suivantes:

pas

vérifier que le système a un GPU compatible cuda

téléchargez et installez la boîte à outils nvidia cuda et cudnn

configuration des variables d'environnement

vérifier l'installation

#

Si vous avez une installation précédente, supprimez-la d'abord.

Sudo apt-get purge nvidia*
Sudo apt remove nvidia-*
Sudo rm /etc/apt/sources.list.d/cuda*
Sudo apt-get autoremove && Sudo apt-get autoclean
Sudo rm -rf /usr/local/cuda*

pour vérifier que votre processeur est cuda, activez la vérification

lspci | grep -i nvidia

le compilateur gcc est requis pour le développement à l'aide de la boîte à outils cuda. pour vérifier la version de l'installation de gcc, entrez

gcc --version

mise à jour du système

Sudo apt-get update
Sudo apt-get upgrade

installer d'autres packages d'importation

Sudo apt-get install g++ freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libglu1-mesa libglu1-mesa-dev

obtenez d'abord le pilote du référentiel PPA

Sudo add-apt-repository ppa:graphics-drivers/ppa
Sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" | Sudo tee /etc/apt/sources.list.d/cuda.list

# installation de CUDA-10.0

Sudo apt-get -o Dpkg::Options::="--force-overwrite" install cuda-10-0 cuda-drivers

configurer vos chemins

echo 'export PATH=/usr/local/cuda-10.0/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
Sudo ldconfig

installer cuDNN v7.5

pour télécharger cuDNN, vous devez être réenregistré ici https://developer.nvidia.com/developer-program/signup

puis téléchargez le formulaire cuDNN v7.5 https://developer.nvidia.com/cudnn

CUDNN_TAR_FILE="cudnn-10.0-linux-x64-v7.5.0.56"
wget https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v7.5.0.56/prod/10.0_20190219/cudnn-10.0-linux-x64-v7.5.0.56.tgz
tar -xzvf ${CUDNN_TAR_FILE}

copiez les fichiers suivants dans le répertoire cuda toolkit.

Sudo cp -P cuda/include/cudnn.h /usr/local/cuda-10.0/include
Sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-10.0/lib64/
Sudo chmod a+r /usr/local/cuda-10.0/lib64/libcudnn*

Enfin, pour vérifier l'installation, vérifiez

nvidia-smi
nvcc -V

installer Tensorflow (un framework d'apprentissage machine open source)

J'ai choisi la version 1.13.1 car elle est stable et compatible avec CUDA 10.0 Toolkit et cuDNN 7.5

pip3 install --user tensorflow-gpu == 1.13.1

Cependant, lorsque je tape nvidia-smi J'obtiens ce qui suit: Échec de l'initialisation de NVML: incompatibilité de version de pilote/bibliothèque

J'ai également essayé de télécharger le dernier pilote NVIDIA (440.35) mais je n'ai pas réussi à exécuter le fichier .run. Il ne cessait de me donner un message disant "Le script de pré-installation fourni par la distribution a échoué!"

Toute aide serait grandement appréciée!

1
bonie10

J'ai enfin pu installer Tensorflow dans ma GeForce RTX 2070 Super avec les installations suivantes:

Ubuntu 18.04

CuDNN 7.5

CUDA 10.0

Pilote NVIDIA version 430.50

Ensuite, je viens de faire pip3 install --user tensorflow-gpu==1.13.1 en utilisant l'environnement Conda.

1
bonie10