Installing tensorflow-GPU
Install tensorflow using pip:
pip3 install –upgrade tensorflow-gpu
If you run import tensorflow
you would get an error like this:
ImportError: Could not find ‘cudart64_90.dll’. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Download and install CUDA 9.0 from this URL: https://developer.nvidia.com/cuda-toolkit
Go to Tensorflow documentation to findout more about requiremnts. As am writting thes the requirements are:
Dobble click cuda_9.0.exe and follow the instructions from the installer to install cuda_9.0
Add cuda path to your global variables system
Add cuDNN v7.0 to the cudas location
For examples on GPU tensorfllow go to https://www.tensorflow.org/guide/using_gpu