这是我的代码,我想使用GPU作为我的代码。目前它正在CPU上运行
elf.graph = tf.Graph()
with self.graph.as_default():
self.face_graph = tf.GraphDef()
fid=tf.gfile.GFile(self.facenet_model, 'rb')
serialized_graph = fid.read()
self.face_graph.ParseFromString(serialized_graph)
tf.import_graph_def(self.face_graph, name='')
self.facenet_sess = tf.Session(graph=self.graph)
self.images_placeholder = self.graph.get_tensor_by_name("input:0")
self.embeddings = self.graph.get_tensor_by_name("embeddings:0")
tensorflow的网站上有一个完整的指南:https://www.tensorflow.org/guide/gpu
确认tf可以看到您的gpu:
查看您的GPU名称:
进行一些手动放置:
把它包起来
相关问题 更多 >
编程相关推荐