擅长:python、mysql、java
<p>在没有看到代码的情况下很难正确回答您的问题。在</p>
<p>下面的代码显示了如何列出可用设备并强制tensorflow使用特定设备。在</p>
<pre><code>def get_available_devices():
local_device_protos = device_lib.list_local_devices()
return [x.name for x in local_device_protos]
get_available_devices()
with tf.device('/gpu:0'):
//Do GPU stuff here
with tf.device('/cpu:0'):
//Do CPU stuff here
</code></pre>