Installing TensorFlow on Pi Supercomputer

· Read in about 1 min · (89 Words)

Login to CentOS 7 node via servers mu06(202.120.58.230) or mu07(202.120.58.231). Then install pre-built TensorFlow from Anaconda.

$ module load gcc/5.4 anaconda/2 cuda/8.0 cudnn/5.1
$ conda create --name deeplearning numpy tensorflow-gpu==1.1.0 keras ipython

Activate the tensorflow environment.

$ source activate deeplearning
$ ipython
> import tensorflow as tf

For GPU-TensorFlow jobs, the following lines are required in the job script:

source /usr/share/Modules/init/bash
unset MODULEPATH
module use /lustre/usr/modulefiles/pi
module purge
module load gcc/5.4 anaconda/2 cuda/8.0 cudnn/5.1
source activate deeplearning

SLURM partitions centos7gpu, centos7k40, centos7k80 and p100 are ready for these jobs.