Premiers pas avec azur

Azure N-series (GPU) : installez CUDA, cudnn, Tensorflow sur UBUNTU 16.04 LTS

Après avoir passé plus de 5 heures, j’ai trouvé cette solution simple :

-Pour vérifier que le système dispose d’un GPU compatible CUDA, exécutez la commande suivante :

lspci | grep -i NVIDIA

Vous verrez une sortie similaire à l’exemple suivant (montrant une carte NVIDIA Tesla K80/M60) :

[![entrez la description de l’image ici][1]][1]

-Désactivation du nouveau pilote :

sudo -i
rmmod nouveau

-Après un reboot : sudo reboot, vérifiez que le pilote est correctement installé en émettant :

lsmod | grep -i nvidia

-Ensuite, téléchargez le package CUDA de Nvidia, …

wget https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64-deb

-… faites savoir qu’apt-get et installez le CUDA Toolkit :

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64-deb
sudo apt-get update
sudo apt-get install -y cuda

-Nous pouvons maintenant vérifier l’état du ou des GPU en exécutant :

nvidia-smi

Ensuite, nous téléchargeons cuDNN…

wget http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz

-… décompressez, copiez la lib64 et incluez les dossiers :

tar -zxf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/lib64/* /usr/local/cuda/lib64/
sudo cp cuda/include/* /usr/local/cuda/include/
sudo rm -R cuda

-Il est temps de nettoyer et de supprimer les archives téléchargées :

rm cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64-deb
rm cudnn-8.0-linux-x64-v5.1.tgz

Pour installer Tensorflow avec CPU/GPU , rendez-vous ici :

https://www.tensorflow.org/install/install_linux#installing_with_anaconda

Référence:

1.https://www.lutzroeder.com/blog/2016-12-27-tensorflow-azure 2.https://www.tensorflow.org/install/install_linux#installing_with_anaconda

[1] : https://i.stack.imgur.com/12dFh.png [2] : https://www.tensorflow.org/install/install_linux#installing_with_anaconda

Azure Powershell →