TensorFlow is an end-to-end open-source platform for machine learning (ML)...

Should you install TensorFlow from pip package manager or conda package manager? Here is a handy installation guide for both, along with several prerequisites.

How to Easily Install TensorFlow through Pip and Conda

Installation Guide of TensorFlow Through Pip and Conda

Although originally developed to resolve large numerical computations, TensorFlow is primarily used for deep learning applications. It aids performance by programming the vital parts in C++ even though it uses Python. With this, there is less human thought overhead with Python and a simpler interface for experimentation can take place. From fast numerical computing to creating deep learning models, this foundation library simplifies the process built on top of TensorFlow.

The platform allows you to take new ideas from concept to code. The simple and flexible architecture enables you to build state-of-the-art models and publications faster. It offers a bundle of workflows with high-level APIs that work wonders for beginners as well as experts.

TensorFlow can help you build ML models with intuitive APIs like Keras. You can efficiently train and deploy these models on-premise, on the cloud, the browser, or on a device, irrespective of the language you use. The models can be trained using high-level APIs that make for easy debugging and immediate model iteration.

TensorFlow competes with other frameworks like Apache MXNet and PyTorch. It can also train and run deep neural networks for word embedding, sequence to sequence models for machine translation, partial differential equation-based simulations, and natural language processing. It offers greater flexibility to train your own models by allowing you to use code from the TensorFlow Model Garden.

As for the working of TensorFlow, it allows developers to create data flow graphs - structures that illustrate how data moves through a series of processing nodes or graphs. The nodes in the graph showcase a mathematical operation and each connection between the nodes is a multidimensional data array which is referred to as a tensor.

A brief on TensorFlow

Before installing TensorFlow, it’s important to check the conditions below.

Prerequisites to install TensorFlow

Here are the following types of installations you can choose to suit your needs.

- Install TensorFlow with GPU support: Although it takes more time to install, the faster processing balances it out. Do make sure that you meet the following checklist before you install TensorFlow with GPU support.

NVIDIA drivers associated with CUDA Toolkit 9.0

GPU card (with CUDA Compute Capability 3.0 or more).

- Install TensorFlow with CPU support: It’s recommended that you use this type of installation only when you don't have an NVIDIA GPU in your system. It will also take less time than the GPU supported version.

What type of installation will suit your requirements the best?

The steps below illustrate how to install TensorFlow without Anaconda.

See what version of Python you have by using the command:

Ensure that the version is Python 3.4+. You can check the same directly using the command below:

If you’re using any other Python version, install the one required for TensorFlow with this command:

$ brew update
$ brew install python # Installs Python 3
$ sudo apt install python3-dev python3-pip

This will help you create the virtual environment required to install TensorFlow. You can do it for macOS and Ubuntu as follows:

$ sudo pip3 install -U virtualenv# system-wide install

Tip: You can use pip or pip3. Use the latter if you have second thoughts. If you still want to use pip, ‌check if it’s possible.

$ pip3 --version
pip 19.1.1 from /Users/inferno/anaconda/lib/python3.6/site-packages/pip (python 3.6)
$ pip --version
pip 19.1.1 from /Users/inferno/anaconda/lib/python3.6/site-packages/pip (python 3.6)

Python without Anaconda

Installing Anaconda offers a simple approach to installing Python. With Anaconda, you will have conda to create a virtual environment to install the package.

Python 3.4+ with Anaconda

TensorFlow will not be loaded on older GPUs when CUDA_FORCE_PTX_JIT=1 is set. This is because packages do not include PTX code except in the latest supported CUDA architecture.

GPU-enabled devices support NVIDIA GPU cards with CUDA architectures of 3.5, 5.0, 6.0, 7.0, 7.5, 8.0 and above.

Hardware requirements

Here is a list of systems that support TensorFlow installation.

macOS 10.12.6 (Sierra) or higher-  (64-bit)

Windows WSL2 Windows 10 19044 or higher - (64-bit).

System requirements

You need to install Miniconda before installing TensorFlow with conda.

curl https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -o 
Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh

Select enter and ‘yes’ during the installation. If needed, restart your terminal or source ~/.bashrc to enable the conda command. Use conda -V to check if Miniconda is installed successfully.

Now that the prerequisites have been fulfilled, let’s get started with the TensorFlow installation process.

Miniconda installation

How to install TensorFlow

We will create a virtual environment in Python where we will have a completely independent set of packages and settings. They won’t conflict with anything in the default local Python environment or other virtual environments.

Tip: Doing this will help you keep different versions of the same package. This means that you can use scikit-learn 0.22 for one project and scikit-learn 0.1 for another.

For developers using Mac/Ubuntu, the command below will create a virtual environment (considering Python without Anaconda).

$ virtualenv --system-site-packages -p python3 tf_2

For developers already using conda, the following command will help create the virtual environment.

Create a Python conda environment

Use the given command in your terminal to activate the virtual environment created above.

For Ubuntu/Mac and Python without Anaconda, use the following command:

For Ubuntu/Mac and Python from Anaconda, use:

Once you activate the virtual environment, the terminal will change to (tf_2) $.

Activate the virtual environment

The GPU and CPU are packed as a single bundle in TensorFlow 2. This means that they don’t need to be installed separately.

The following pip command will help you install the latest version of TensorFlow in the newly created conda environment above.

# Requires the latest pip
pip install --upgrade pip


# Current stable release for CPU and GPU
pip install tensorflow

Note: If you want to install the current tf-nightly build, the command below will help you do so:

Install TensorFlow 2

Once TensorFlow is installed, check the version of the tf package.

(tf_2) $ python -c "import tensorflow as tf; x = [[2.]]; print('tensorflow version', tf.__version__);
 print('hello, {}'.format(tf.matmul(x, x)))"

You can expect the following output with successful installation:

tensorflow version 2.0.0-rc1
hello, [[4.]]

Test the installation

Here are a few quick versions of the install commands for various systems.

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
python3 -m pip install tensorflow
# Verify install:
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/
python3 -m pip install tensorflow
# Verify install:
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

python3 -m pip install tensorflow
# Verify install:
python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

python3 -m pip install tf-nightly
# Verify install:
python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

Quick install commands

Step 1: Begin by finding if the GPU is compatible with TensorFlow GPU.

Step 2: Download and install the CUDA toolkit.

Step 3: Sign up on the NVIDIA Developer website and download cuDNN.

Step 4: Install cuDNN by extracting the contents into the toolkit path installed in step 2.

Note: Replace the files you have in the CUDA Toolkit Directory.

Step 5: Check your path variables. Look for CUDA_HOME. If it is not present, add it manually.

Step 6: Check the path variables’ presence in the toolkit paths.

Step 8: Create an environment with Python and pip packages installed.

Step 9: Install by “pip install TensorFlow-gpu” and test it.

In this article, we examined the TensorFlow installation process in detail with step-by-step instructions. Now that you’ve learned how to install it through pip and conda, you can use it for your machine learning projects.

Checklist

TensorFlow is an end-to-end open-source platform for machine learning (ML), backed by a comprehensive yet flexible ecosystem of tools, libraries, and communities. It is that magic well that allows developers to build and deploy ML-powered applications easily. This free software library is used for a variety of tasks with a focus on the inference and training of deep neural networks. In this article, we’ll look at how to install TensorFlow through pip and conda as well as the prerequisites.

To install TensorFlow on Anaconda pip, follow these steps: Open command prompt. If you are operating only on one version of Python, type in cmd: C:/>conda install tensorflow. If you have multiple versions of Python, look for the version you want to install TensorFlow. Then, type in cmd: C:/>conda install tensorflow python=version(e.g.python=3.5).

The fundamental difference between conda and pip is what lies inside the packages. conda packages include C libraries, Python libraries, executables, and even the Python interpreter itself. On the other hand, pip packages are Python libraries like Matplotlib and NumPy.

Anaconda includes the most common packages or tools that are required by data scientists. They are essentially the hardware store of the tools required in data science. On the other hand, conda is a package manager that handles installation, updates, and removal of different packages.

Frequently Asked Questions

What’s up with Turing?
Get the latest news about us here.

Press

Know more about remote work.
Checkout our blog here.

Blog

Have any questions?
We’d love to hear from you.

FOR DEVELOPERS

Installation Guide of TensorFlow Through Pip and Conda

Author

Srishti Chaudhary

A Step-by-Step Guide for Binary Image Classification in TensorFlow: Detection of Pneumothorax From Chest X-ray

What Is the Necessity of Bias in Neural Networks?

How to Make a Simplified Multilayer Perceptron in TensorFlow

In-Depth Guide on Deep Learning Frameworks - Keras, TensorFlow, PyTorch

PyTorch vs TensorFlow: What is Best for Deep Learning?

Frequently Asked Questions

Press

Blog

Contact

Hire remote developers