How To Install Tensorflow 2.5 using conda?
Posted by TechGopal on March 30, 2021 at 6:10 pm
This is a guide to install TensorFlow 2.5 with the following dependency versions
– Python : 3.8
– CUDA Toolkit 11.0
– cuDNN 8.0
1. Create a new conda envirniment with python 3.8.
conda create -n tensorflow_25 python=3.8
We need to switch the newly created enviroment.
conda activate tensorflow_25
2. Install CUDA Toolkit
conda install -c anaconda cudatoolkit
3. Install cuDNN
conda install -c conda-forge cudnn
4. Install Tensorflow GPU
pip install tensorflow==2.5
5. Verify installation
python
This will open the python terminal. Use the following command to check if GPU is detected.
print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))