View source for ML computers tensorflow


==General info==
*Install the package tensorflow-gpu for GPU acceleration.
*For tensorflow-gpu <= 1.12 you need  CUDA9.0  TF1.13,CUDA10.0)
*For tensorflow-gpu >= 1.13 you need  CUDA10.0

*By default tensorflow installs the cpu version. For being sure the GPU is being used, run these lines with python3:
 from tensorflow.python.client import device_lib  
 print(device_lib.list_local_devices())

*The GPU should appear as a device.

==Setting up tensorflow-gpu on Ubuntu 16.04 for python3==

*First we install pip3: [[ML computers python|python3 and pip3 installation]]

*Then we do:
 pip3 install  tensorflow-gpu==1.2.0

*If conflict with setuptools version
*solution:  
 pip3 install --upgrade setuptools==41.0.0
 sudo pip3 install --upgrade tensorflow-gpu==1.2.0

*How to check the version installed:
 pip3 show tensorflow-gpu

*Problem: After installing tensorflow jupyter notebook was not working. 
*Solution: 
 pip3 uninstall pyzmq
 pip3 install pyzmq
*More info: https://github.com/jupyter/notebook/issues/3435

*Problem: Error -> ImportError: No module named model_utils
*Solution:
 pip3 list
*and installing the same version of  tensorflow-estimator than of tensorflow-gpu:
 pip3 uninstall tensorflow-estimator    
 pip3 install tensorflow-estimator==1.14.0
[[Category:AI & Machine Learning]]