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

save_dir

Results

Add cuda path to your global variables system

Results

Add cuDNN v7.0 to the cudas location

For examples on GPU tensorfllow go to https://www.tensorflow.org/guide/using_gpu

Manuel Cuevas

Manuel Cuevas

Hello, I'm Manuel Cuevas a Software Engineer with background in machine learning and artificial intelligence.

Recent post