Open dathayer23 opened 6 years ago
BTW have installed tensorflow-gpu on my anaconda tensorflow instance. Everything installed successfully.
Hey, sorry it took me so long to see this.
If you have 2 gpus you should use the dist-version
otherwise it will only use one of the gpus. You shouldn't have to specifically import tensorflow-gpu in the code though. Do you know if it ran on any of your cards?
You should see something like this with you card listed:
Creating TensorFlow device (/device:GPU:0) -> (device: 0, name: Tesla P100 ...)
I have installed tensorflow-gpu on my desktop which has two high powered NVidia gpu cards and it unfortunately is not using the gpu. I have implicitly told it to import tensorflow-gpu and it causes an error. Does anyone know how to get the main.py code to use the GPU?