Changes

Jump to navigation Jump to search
==Installing Tensorflow as root for All users==
'''Important note: Currently on DB Server, pip/pip3 is working with Python3.6 rather than Python3. Hence the following installs a copy of tensorflow-gpu 1.9.0 for Python3.6 for all users'''
 
==Installation==
*0. Deleted previously installed tensorflow with CPU support:
sudo pip3 uninstall tensorflow
*1. Used this command to install tensorflow-gpu:
sudo pip3 install -U tensorflow-gpu
 
==Path variable (crucial)==
If you logged on as a user using tensorflow for the first time, you need to set the CUDA Toolkit 9.0 environment variables. Type into terminal
nano .bashrc
 
Add the following:
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64\
${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
 
Save and exit (CTRL + O and CTRL + X). Type
source .bashrc
===Testing Tensorflow with GPU as (non-root) user===

Navigation menu