擅长:python、mysql、java
<p>您可以将其放在模型函数的开头,即定义模型时,应编写:</p>
<pre><code>def cnn_model_fn2(...):
with tf.device('/gpu:0'):
...
</code></pre>
<p>不过,我希望tensorflow会自动为您的模型使用gpu。您可能需要检查是否正确检测到它:</p>
<pre><code>from tensorflow.python.client import device_lib
device_lib.list_local_devices()
</code></pre>