TensorFlow is a deep learning framework that provides an easy interface to a variety of functionalities, required to perform state-of-the-art deep learning tasks such as image recognition, text classification, and so on. It is a machine learning framework developed by Google and is used for designing, building, and training of deep learning models such as neural networks.
When installing TensorFlow, you can choose either the CPU-only or GPU-supported version. You must install the following software in order to run the GPU version of TensorFlow:
- NVIDIA GPU drivers
- CUDA Toolkit: CUDA 9.0.
- NCCL 2.2 (optional)
- cuDNN SDK (7.2 or higher)
- TensorRT for improved latency and throughput.
Use can use the following code to install TensorFlow in Python 2.7:
pip install tensorflow
This may take a few minutes. After the installation is completed, just try to import the tensorflow module. If no error is encountered, that means the installation is successfull.