J'ai essayé d'installer TensorFlow (version GPU) sur Ubuntu

Aperçu

Un enregistrement de l'installation de TensorFlow, une bibliothèque d'intelligence artificielle open source publiée par Google, sur Ubuntu. Jusqu'à ce que vous exécutiez la formation CIFAR-10 avec CUDA Enable.

Officiel: http://www.tensorflow.org Git: https://tensorflow.googlesource.com/tensorflow

Configuration de la machine

Procédure d'installation

Fondamentalement, c'est comme Officiel, mais enregistrez-le dans l'ordre dans lequel vous êtes allé.

(1). Récupérez l'arborescence des sources de Git

#Uniquement pour ceux qui n'ont pas Git
$ sudo apt-get install git

$ git clone --recurse-submodules https://github.com/tensorflow/tensorflow

(2). Installation de Cuda Toolkit 7.0
Puisque la version 7.5 est restée bloquée dans les étapes suivantes, installez la version 7.0
DL et installez la version du programme d'installation réseau Ubuntu 14.04 DEB (10 Ko) (cuda-repo-ubuntu1404-7-0-local_7.0-28_amd64.deb) à partir de la page suivante
https://developer.nvidia.com/cuda-toolkit-70

$ sudo dpkg -i cuda-repo-ubuntu1404-7-0-local_7.0-28_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda-7-0 

(3). Installation de CUDNN Toolkit 6.5
CUDNN DL doit être enregistré sur le site de Nvidia (et j'ai l'impression d'avoir dû attendre quelques jours pour terminer l'inscription)
DL et installez la bibliothèque cuDNN v2 pour Linux (cudnn-6.5-linux-x64-v2.tgz) à partir de
https://developer.nvidia.com/rdp/cudnn-archive

$ tar xvzf cudnn-6.5-linux-x64-v2.tgz 
$ sudo cp cudnn-6.5-linux-x64-v2/cudnn.h /usr/local/cuda/include
$ sudo cp cudnn-6.5-linux-x64-v2/libcudnn* /usr/local/cuda/lib64

** Redémarrez ici **

(4). Installation de VirtualEnv et création d'un conteneur

#Installation
$ sudo apt-get install python-pip python-dev python-virtualenv

#Création de conteneurs
$ virtualenv --system-site-packages ~/tensorflow-GPU

Modifier ~ / tensorflow-GPU / bin / activate Ajoutez les deux lignes suivantes à la fin

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"
export CUDA_HOME=/usr/local/cuda

(5). Installation de TensorFlow
Exécutez ce qui suit une seule fois lorsque le chemin de la bibliothèque CUDA change

#(1)Déplacer vers le répertoire source obtenu dans
$ cd ~/tensorflow/tensorflow

#Autorisation effective
$ chmod x+ ./configure

$ ./configure
Do you wish to bulid TensorFlow with GPU support? [y/n] y
GPU support will be enabled for TensorFlow

Please specify the location where CUDA 7.0 toolkit is installed. Refer to
README.md for more details. [default is: /usr/local/cuda]: /usr/local/cuda

Please specify the location where CUDNN 6.5 V2 library is installed. Refer to
README.md for more details. [default is: /usr/local/cuda]: /usr/local/cuda

Setting up Cuda include
Setting up Cuda lib64
Setting up Cuda bin
Setting up Cuda nvvm
Configuration finished

Activer le conteneur À l'avenir, si vous souhaitez lancer un nouveau terminal et travailler avec le conteneur tensorflow-GPU, activez-le d'abord ci-dessous

$ cd ~/tensorflow-GPU
$ source bin/activate

Installez TensorFlow pour GPU

(tensorflow-GPU) $ pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.5.0-cp27-none-linux_x86_64.whl

Essayez de bouger

(1). MNIST

Lorsque j'ai essayé d'exécuter MNIST, j'ai eu une erreur (au 15 novembre 2015), alors apportez les modifications suivantes

(tensorflow-GPU) $ cd ~/tensorflow/tensorflow/g3doc/tutorials/mnist/
#Renommez le fichier à remplacer
(tensorflow-GPU) $ mv mnist.py mnist_org.py
#Récupérez l'ancienne version du référentiel
(tensorflow-GPU) $ wget https://raw.githubusercontent.com/tensorflow/tensorflow/1d76583411038767f673a0c96174c80eaf9ff42f/tensorflow/g3doc/tutorials/mnist/mnist.py

Les lignes 23 et 24 de Fully_connected_feed.py devraient ressembler à ceci:

#from tensorflow.g3doc.tutorials.mnist import input_data
#from tensorflow.g3doc.tutorials.mnist import mnist
import input_data
import mnist

Essayez de bouger

(tensorflow-GPU) $ python fully_connected_feed.py
Extracting data/train-images-idx3-ubyte.gz
Extracting data/train-labels-idx1-ubyte.gz
Extracting data/t10k-images-idx3-ubyte.gz
Extracting data/t10k-labels-idx1-ubyte.gz
I tensorflow/core/common_runtime/local_device.cc:25] Local device intra op parallelism threads: 8
I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:888] successful NUMA node read from SysFS had negative value (-1), but there must be at least one NUMA node, so returning NUMA node zero
I tensorflow/core/common_runtime/gpu/gpu_init.cc:88] Found device 0 with properties: 
name: GeForce GTX 970
major: 5 minor: 2 memoryClockRate (GHz) 1.253
pciBusID 0000:01:00.0
Total memory: 4.00GiB
Free memory: 3.22GiB
I tensorflow/core/common_runtime/gpu/gpu_init.cc:112] DMA: 0 
I tensorflow/core/common_runtime/gpu/gpu_init.cc:122] 0:   Y 
I tensorflow/core/common_runtime/gpu/gpu_device.cc:643] Creating TensorFlow device (/gpu:0) -> (device: 0, name: GeForce GTX 970, pci bus id: 0000:01:00.0)
I tensorflow/core/common_runtime/gpu/gpu_region_allocator.cc:47] Setting region size to 3144105984
I tensorflow/core/common_runtime/local_session.cc:45] Local session inter op parallelism threads: 8
Step 0: loss = 2.34 (0.300 sec)
Step 100: loss = 2.13 (0.002 sec)
Step 200: loss = 1.90 (0.002 sec)
Step 300: loss = 1.52 (0.002 sec)
Step 400: loss = 1.22 (0.002 sec)
Step 500: loss = 0.84 (0.002 sec)
Step 600: loss = 0.82 (0.002 sec)
Step 700: loss = 0.68 (0.002 sec)
Step 800: loss = 0.71 (0.002 sec)
Step 900: loss = 0.51 (0.002 sec)
Training Data Eval:
  Num examples: 55000  Num correct: 47651  Precision @ 1: 0.8664
Validation Data Eval:
  Num examples: 5000  Num correct: 4363  Precision @ 1: 0.8726
Test Data Eval:
  Num examples: 10000  Num correct: 8745  Precision @ 1: 0.8745
Step 1000: loss = 0.46 (0.002 sec)
Step 1100: loss = 0.44 (0.038 sec)
Step 1200: loss = 0.52 (0.002 sec)
Step 1300: loss = 0.43 (0.002 sec)
Step 1400: loss = 0.64 (0.002 sec)
Step 1500: loss = 0.34 (0.002 sec)
Step 1600: loss = 0.41 (0.002 sec)
Step 1700: loss = 0.34 (0.002 sec)
Step 1800: loss = 0.30 (0.002 sec)
Step 1900: loss = 0.35 (0.002 sec)
Training Data Eval:
  Num examples: 55000  Num correct: 49286  Precision @ 1: 0.8961
Validation Data Eval:
  Num examples: 5000  Num correct: 4529  Precision @ 1: 0.9058
Test Data Eval:
  Num examples: 10000  Num correct: 9012  Precision @ 1: 0.9012

(2). CIFAR-10 Essayez de bouger

(tensorflow-GPU) $ cd ~/tensorflow/tensorflow/models/image/cifar10/
(tensorflow-GPU) $ python cifar10_train.py
>> Downloading cifar-10-binary.tar.gz 100.0%
Succesfully downloaded cifar-10-binary.tar.gz 170052171 bytes.
Filling queue with 20000 CIFAR images before starting to train. This will take a few minutes.
I tensorflow/core/common_runtime/local_device.cc:25] Local device intra op parallelism threads: 8
I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:888] successful NUMA node read from SysFS had negative value (-1), but there must be at least one NUMA node, so returning NUMA node zero
I tensorflow/core/common_runtime/gpu/gpu_init.cc:88] Found device 0 with properties: 
name: GeForce GTX 970
major: 5 minor: 2 memoryClockRate (GHz) 1.253
pciBusID 0000:01:00.0
Total memory: 4.00GiB
Free memory: 3.20GiB
I tensorflow/core/common_runtime/gpu/gpu_init.cc:112] DMA: 0 
I tensorflow/core/common_runtime/gpu/gpu_init.cc:122] 0:   Y 
I tensorflow/core/common_runtime/gpu/gpu_device.cc:643] Creating TensorFlow device (/gpu:0) -> (device: 0, name: GeForce GTX 970, pci bus id: 0000:01:00.0)
I tensorflow/core/common_runtime/gpu/gpu_region_allocator.cc:47] Setting region size to 3120906240
I tensorflow/core/common_runtime/local_session.cc:45] Local session inter op parallelism threads: 8
2015-11-17 02:14:46.611756: step 0, loss = 4.68 (6.9 examples/sec; 18.481 sec/batch)
2015-11-17 02:14:49.068440: step 10, loss = 4.65 (562.6 examples/sec; 0.228 sec/batch)
2015-11-17 02:14:51.224980: step 20, loss = 4.65 (617.0 examples/sec; 0.207 sec/batch)
2015-11-17 02:14:53.375918: step 30, loss = 4.62 (664.1 examples/sec; 0.193 sec/batch)
2015-11-17 02:14:55.513463: step 40, loss = 4.60 (610.3 examples/sec; 0.210 sec/batch)
2015-11-17 02:14:57.696431: step 50, loss = 4.58 (615.1 examples/sec; 0.208 sec/batch)
2015-11-17 02:14:59.877955: step 60, loss = 4.57 (567.3 examples/sec; 0.226 sec/batch)
2015-11-17 02:15:02.101614: step 70, loss = 4.55 (621.1 examples/sec; 0.206 sec/batch)
2015-11-17 02:15:04.593141: step 80, loss = 4.52 (490.3 examples/sec; 0.261 sec/batch)
2015-11-17 02:15:06.983452: step 90, loss = 4.52 (641.4 examples/sec; 0.200 sec/batch)
2015-11-17 02:15:09.232584: step 100, loss = 4.50 (563.8 examples/sec; 0.227 sec/batch)
2015-11-17 02:15:11.783752: step 110, loss = 4.48 (538.0 examples/sec; 0.238 sec/batch)
2015-11-17 02:15:13.997070: step 120, loss = 4.46 (589.4 examples/sec; 0.217 sec/batch)
2015-11-17 02:15:16.458028: step 130, loss = 4.45 (467.8 examples/sec; 0.274 sec/batch)
2015-11-17 02:15:19.128071: step 140, loss = 4.42 (581.1 examples/sec; 0.220 sec/batch)
2015-11-17 02:15:21.491835: step 150, loss = 4.40 (568.2 examples/sec; 0.225 sec/batch)
2015-11-17 02:15:23.962043: step 160, loss = 4.39 (635.4 examples/sec; 0.201 sec/batch)
...

À propos, si vous l'exécutez sur la version du processeur, il faut environ deux fois plus de temps pour prendre un lot, il semble donc qu'il soit accéléré en utilisant le GPU.

Créer tutorials_example_trainer pour GPU

Installez bazel dans un shell normal au lieu d'un environnement de conteneur

#Installation des packages requis
$ sudo add-apt-repository ppa:webupd8team/java
$ sudo apt-get update
$ sudo apt-get install oracle-java8-installer
$ sudo apt-get install pkg-config zip g++ zlib1g-dev unzip

#Installateur DL bazel
$ wget https://github.com/bazelbuild/bazel/releases/download/0.1.1/bazel-0.1.1-installer-linux-x86_64.sh

#Installation
$ chmod +x bazel-0.1.1-installer-linux-x86_64.sh 
$ ./bazel-0.1.1-installer-linux-x86_64.sh --user

Modifiez ~ / .bashrc et ajoutez ce qui suit à la fin

export PATH="$PATH:$HOME/bin"

Créez un fichier dans le chemin ci-dessous ~/tensorflow/third_party/gpus/cuda/cuda.config

Le contenu du fichier est le suivant

CUDA_TOOLKIT_PATH="/usr/local/cuda"
CUDNN_INSTALL_PATH="/usr/local/cuda"

Il devrait y avoir un lien symbolique / usr / local / cuda (=> / usr / local / cuda-7.0) s'il est installé avec les paramètres par défaut Si vous avez modifié le chemin d'installation, modifiez-le en conséquence

Exécutez ./configure avant de construire

$ cd ~/tensorflow-GPU
$ source bin/activate
(tensorflow-GPU) $ cd ~/tensorflow
(tensorflow-GPU) $ ./configure
Do you wish to bulid TensorFlow with GPU support? [y/n] y
GPU support will be enabled for TensorFlow

Please specify the location where CUDA 7.0 toolkit is installed. Refer to
README.md for more details. [default is: /usr/local/cuda]: /usr/local/cuda

Please specify the location where CUDNN 6.5 V2 library is installed. Refer to
README.md for more details. [default is: /usr/local/cuda]: /usr/local/cuda

Setting up Cuda include
Setting up Cuda lib64
Setting up Cuda bin
Setting up Cuda nvvm
Configuration finished

Construire

(tensorflow-GPU) $ bazel build -c opt --config=cuda tensorflow/cc:tutorials_example_trainer

Il a fallu environ 10 minutes pour terminer la construction

Recommended Posts

J'ai essayé d'installer TensorFlow (version GPU) sur Ubuntu
J'ai installé tensorRT sur Ubuntu 18.04
J'ai installé ROS sur Ubuntu 18.04
RTKLIB 2.4.3 Version GUI de la version CLI installée sur Ubuntu 18.04
Installez TensorFlow sur Ubuntu
[Ubuntu 18.04] Construction de l'environnement Tensorflow 2.0.0-GPU
Présentation de TensorFlow à Ubuntu + Python 2.7
63e jour, j'ai installé tensorflow.
Exécutez la version GPU tensorflow sur une instance ponctuelle d'AWS EC2
Installez Caffe sur Ubuntu 14.04 (GPU)
Construction de l'environnement Ubuntu14.04 + GPU + TensorFlow
J'ai installé Ubuntu sur une mémoire USB sur un PC à double démarrage
Je suis tombé sur TensorFlow (Quelle est la mémoire du GPU)
Installez CUDA10.1 + cuDNN7.6.5 + tensorflow-2.3.0 sur Ubuntu 18.04
Supprimer Ubuntu installé sur une machine Windows 10
J'ai essayé Cython sur Ubuntu sur VirtualBox
Je ne pouvais pas entrer le japonais sur Ubuntu 20.04
J'ai essayé d'installer Linux sur mon Mac
Exécutez TensorFlow sur une instance GPU sur AWS
Je suis tombé sur l'installation de la phrase sur ubuntu
J'ai essayé d'installer Kivy dans un environnement Mac
Installez Pleasant sur Ubuntu 20.04 (version .NetCore3.1 / PostgreSQL)
J'ai construit un environnement TensorFlow avec windows10
J'étais accro à l'exécution de tensorflow sur GPU avec le pilote NVIDIA 440 + CUDA 10.2
Version de juin 2017 pour créer un environnement Tensorflow / Keras sur une instance GPU d'AWS
Essayez Tensorflow avec une instance GPU sur AWS
Shebang sur Ubuntu 20.04
Utilisez la dernière version de PyCharm sur Ubuntu
J'ai essayé la détection d'objets avec YOLO v3 (TensorFlow 2.1) sur le GPU de windows!
Remarques sur l'utilisation de TensorFlow sur Bash sur Ubuntu sous Windows
Installez la version 64 bits d'Ubuntu 20.04 LTS (serveur) sur RaspberryPi3B +
Exécutez Tensorflow à partir de Jupyter Notebook sur Bash sur Ubuntu sous Windows
J'ai un UnicodeDecodeError lorsque je pip installé sur ubuntu
Je souhaite utiliser OpenJDK 11 avec Ubuntu Linux 18.04 LTS / 18.10
Installez Tensorflow sur Mac
Création d'un environnement d'apprentissage automatique à partir de zéro (version windows10 + Anaconda + VSCode + Tensorflow + GPU)
[Streamlit] Je l'ai installé
Installez JModelica sur Ubuntu
Activer le GPU pour tensorflow
Je souhaite utiliser l'environnement de bureau d'Ubuntu sur Android pour le moment (version Termux)
Construire TensorFlow sous Windows
construire Python sur Ubuntu
Installez Python 3.3 sur Ubuntu 12.04
Installation de pyenv sur ubuntu 16.04
Installez Theano sur Ubuntu 12.04
Installez angr sur Ubuntu 18.04
J'ai essayé d'exécuter TensorFlow
Installez pip / pip3 sur Ubuntu
J'ai installé la bibliothèque de machine learning automatique auto-sklearn sur centos7
J'ai fait un modèle VGG16 en utilisant TensorFlow (en chemin)