TensorFlow gpu java,CPU和GPU Tensorflow安装

I'm very new with TensorFlow.

I want to run my code on my CUDA gpu. So I've installed TensorFlow -gpu, after I've installed normal TensorFlow.

How can I tell Python that it takes gpu based TensorFlow?

解决方案

If you have tensorflow-gpu installed there really isn't any reason to also have tensorflow. Without the presence of a gpu it will just run on cpu anyway.

To be specific in which GPU you use:

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "0"

in place of "0" you can either list GPUs (if you have multiple), or "" if you want it to run on cpu.

alternatively, specify in the session:

sess = tf.Session(config=tf.ConfigProto(device_count={'GPU': 0}))

Furthermore you can check which version your computer prioritizes by opening python console and typing:

>>> import tensorflow

>>> tensorflow

'/home/.../python3.6/site- packages/tensorflow/__init__.py'>

^

|

here