capsule-networks icon indicating copy to clipboard operation
capsule-networks copied to clipboard

Published 20 hours ago • verified publisher icon bourdakos1

tensorflow-gpu not working.

Open dathayer23 opened this issue 7 years ago • 2 comments

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?

dathayer23 avatar Jan 31 '18 00:01 dathayer23

BTW have installed tensorflow-gpu on my anaconda tensorflow instance. Everything installed successfully.

dathayer23 avatar Jan 31 '18 00:01 dathayer23

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 ...)

bourdakos1 avatar Feb 15 '18 16:02 bourdakos1