I can't import tensorflow-gpu

Question

I've successfully installed tensorflow with pip install tensorflow and that all works as expected.

I can also successfully install tensorflow-gpu with pip install tensorflow-gpu but I can't import it in my python script:

import tensorflow-gpu

File "<stdin>", line 1
import tensorflow-gpu
                 ^
SyntaxError: invalid syntax

i've installed CUDA v9.0 and run windows 10

Answer 1

To ensure that the tensorflow package is using your GPU, do this:

import tensorflow as tf

sess = tf.Session()

See the output on the console, if it shows your GPU information on creation of the session as shown below.

Notice "GeForce 940MX" in the information. Also note that Tensorflow will use a Nvidia GPU only if the compute capability score is above 3.5 . More about that here.

If it's not using the GPU, then it won't output GPU information, it'll just show something similar to this:

Answer 2

The package on pypi is called tensorflow-gpu but you just import it with "tensorflow"

 import tensorflow as tf