I have a good configuration GPU on which I used to play FIFA. Switching to AI, I wanted to use GPU for Deep Learning instead of playing games. But….. I was scared of Tensorflow installations with incompatible Cuda Versions. In this article I will explain the conventional approach and the new optimized approach and why we should dump pip and use conda instead.
To install Tensorflow for GPU I had to follow the steps(Steps are for windows) :
- First find if the GPU is compatible with Tensorflow GPU or not! (From Here)
- Download and Install Cuda Toolkit from here.
- Download cuDNN by signing up on Nvidia Developer Website
- Install cuDNN by extracting the contents of cuDNN into the Toolkit path installed in Step 2. There will be files that you have to replace in CUDA Toolkit Directory.
- Is that it? No, then you need to check your path variables if CUDA_HOME is present or not. If not, please add it manually.
- Then check the path variables if your toolkit paths are available or not.
- Then finally install Anaconda or Miniconda
- Creating an Environment with Python and Pip packages installed.
- Then finally ‘pip install tensorflow-gpu’.
- Test your installation.
Well you will say that’s easy right? Why not go ahead and try it. Remember I am not giving visuals for any of the steps above because these are just for your knowledge and to explain the problem we had with this process.
There is a probability of 1% that this process will go right for you!
Because of the version numbering
Different Versions of Tensorflow support different cuDNN and CUDA Verisons (In this table CUDA has an integer value but when you go to download it is actually a float which makes numbering and compatibility more difficult). Also cuDNN and conda were not a part of conda.
The Award Winning New Approach
Install Miniconda or Anaconda and then run this command.
conda create --name tf_gpu tensorflow-gpu
Well is that it? YES.
This command will create an environment first named with ‘tf_gpu’ and will install all the packages required by tensorflow-gpu including the cuda and cuDNN compatible verisons.
And also it will not interfere with your current environment all ready set up.
If you have a hard time visualizing the command I will break this command into three commands.
conda create --name tf_gpu
conda install tensorflow-gpu
Note: This works for Ubuntu users as well. No more long scripts to get the DL running on GPU.
Testing your Tensorflow Installation
To test your tensorflow installation follow these steps:
- Open Terminal and activate environment using ‘activate tf_env’.
- Go to python console using ‘python’
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
Source: Deep Learning on Medium