Tensorflow

Article mis en ligne le 20 février 2019

dernière modification le 26 février 2021

CALMIP recommande l’utilisation plutôt de conda pour gérer ses environnements python

La bibliothèque TensorFlow est accessible au travers du module python/3.6.3.

Environnement Python + CUDA

module purge
module load python/3.6.3 cuda/9.0.176.2

Lancement de TensorFlow avec Python

[login@olympelogin1 ~]$ python
Python 3.6.3 |Intel Corporation| (default, Feb 12 2018, 06:37:09) 
[GCC 4.8.2 20140120 (Red Hat 4.8.2-15)] on linux
>>> import tensorflow as tf
>>> tf.__version__
'1.11.0'

Pour installer la dernière version de tensorflow dans son espace utilisateur :

module purge
module load python/3.7.4
module load cuda/10.1.105
pip install tensorflow --user

python
Python 3.7.4 (default, Nov 22 2019, 21:31:39) 
[GCC 7.3.0] :: Intel(R) Corporation on linux
Type "help", "copyright", "credits" or "license" for more information.
Intel(R) Distribution for Python is brought to you by Intel Corporation.
Please check out: https://software.intel.com/en-us/python-distibution
>>> import tensorflow as tf
>>> tf.__version__
'2.2.0'

Vérifier que son installation est fonctionnelle

Vous pouvez utiliser le script ci-joint pur vérifier que votre installation de python 2 fonctionne correctement sur les GPUs :

Copiez le script dans un répertoire sur Olympe
Suivez les instructions en début de script (en Anglais)

Si vous ’avez pas assez de place dans votre espace utilisateur pour réaliser l’installation, vous pouvez demander une augmentation de quota à support.calmip@univ-toulouse.fr

La bibliothèque TensorFlow est accessible au travers du module python/3.6.3.

Pour installer la dernière version de tensorflow dans son espace utilisateur :

Vérifier que son installation est fonctionnelle

Dans la même rubrique