TensorFlow is a symbolic math library popular for machine learning applications.
Installing via PowerAI¶
This is the preferred method to install TensorFlow. See the Installing PowerAI for more information.
Installing via "free" channel¶
' Due to the removal of the free channel from the default conda configuration and certain problems that can arise during updates, this method is no longer recommended. Users should instead make use of PowerAI.'
TensorFlow can be installed quickly by using conda. CCI recommends using the DCS system for TensorFlow. It is recommended to have only one common TensorFlow install for each CCI project to save on disk quota space. This can be achieved by installing to the barn-shared directory and setting the necessary permissions for group access.
To access software installers from the internet you need to enable the CCI proxy server in your shell environment. Instructions can be found on the Proxy page.
The channel containing tensorflow-gpu was removed from the default
list in conda 4.7. If you are using conda 4.7 (or newer) please make
sure you add this channel:
conda config --add channels https://repo.anaconda.com/pkgs/free. You
can check to see if this channel is enabled and the version of conda
conda info. If you upgrade from 4.6 to 4.7 then the channel will
be removed and you will need to add it again.
When using the DCS system, it is imperative that an appropriate version
of CUDA is selected by conda. At the time of writing, directly
tensorflow-gpu package will select a version of CUDA
that does not work. Install
cudatoolkit first and ensure it is at
least version 9.0:
conda install -y cudatoolkit
conda install -y tensorflow-gpu
Many users may find they also require keras and a GPU-variant can also be installed by conda:
conda install -y keras-gpu