【发布时间】：2017-05-17 20:54:01
【问题描述】：
（没有愚蠢的问题，但有很多好奇的白痴[我]，如果答案很明显，我很抱歉）。我有一个 mac，我想在 gpu 上运行一个张量流程序。可悲的是，我的电脑没有必要的 gpu，所以我希望使用亚马逊中的一个。我成功创建了一个 p2.xlarge 实例，但我不知道如何从我的程序中实际访问该 gpu。程序如何知道使用亚马逊 GPU？我需要包含一些代码吗？这是我目前拥有的代码，打算在本地 gpus 上使用：
类 DeviceCellWrapper(tf.contrib.rnn.GRUCell): def init(self,device,cell): self._cell = 细胞 self._device = 设备
@property
def state_size(self):
return self._cell.state_size
@property
def output_size(self):
return self._cell.output_size
def __call__(self,inputs,state,scope=None):
with tf.device(self._device):
return self._cell(inputs,state,scope)
设备 = ["/gpu:0]
那么我应该用什么替换“/gpu:0”？
【问题讨论】：
-
它会在可用时自动使用 GPU。 IE
sess.run(tf.ones(1000)*2)将在 GPU 上做乘法 -
好的，那么我应该怎么做才能使 GPU 可用？只运行实例就足够了吗？
标签： amazon-web-services amazon-ec2 tensorflow amazon