How to install tensorflow in a conda environment

In this post, we will learn how to install tensorflow 2 in a conda environment. I would be installing tensorflow in two steps, first we will create a conda python environment and then install the tensorflow 2 using pip

Ensure that the anaconda is installed properly in your mac/windows laptop and path is setup and conda commands can be accessed from your windows cmd or mac terminals

Step 1: Create python conda environemt

A conda environment with a specific python version 3.7 can be created using the following command

conda create -n tf-venv python=3.7

Here tf-env is my tensorflow environment name, this is a user-defined name

Activate this environment by typing below command in your terminal

conda activate tf-venv

Step2 :Install tensorflow 2 in conda environment

In tensorflow 2, CPU and GPU are packaged as one single bundle so you don’t have to install it separately.

we can use the following pip command to install the latest tensorflow version in the conda environment created above

# Requires the latest pip
pip install --upgrade pip

# Current stable release for CPU and GPU
pip install tensorflow

if you want to install the current tf-nightly build then use the below command

pip install tf-nightly

Once tensorflow is installed then check the version of the tf package, As of writing this article the latest tensorflow version was 2.7.0

>>> import tensorflow as tf
>>> tf.__version__
'2.7.0'