Issue
How do I use TensorFlow GPU version instead of CPU version in Python 3.6 x64?
import tensorflow as tf
Python is using my CPU for calculations.
I can notice it because I have an error:
Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
I have installed tensorflow and tensorflow-gpu.
How do I switch to GPU version?
Solution
Follow this tutorial Tensorflow GPU I did it and it works perfect.
Attention! - install version 9.0! newer version is not supported by Tensorflow-gpu
Steps:
- Uninstall your old tensorflow
- Install tensorflow-gpu
pip install tensorflow-gpu
- Install Nvidia Graphics Card & Drivers (you probably already have)
- Download & Install CUDA
- Download & Install cuDNN
- Verify by simple program
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
Answered By - Ashwel
0 comments:
Post a Comment
Note: Only a member of this blog may post a comment.