search

bourdakos1 / capsule-networks

A Tensorflow implementation of Capsule Networks
MIT License
478 stars 156 forks source link

tensorflow-gpu not working. #3

Open dathayer23 opened 6 years ago

dathayer23 commented 6 years ago

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 commented 6 years ago

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

bourdakos1 commented 6 years ago

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