tensorflow下设置使用某一块GPU、多GPU、CPU的情况

MachineLP

发布于 2018-01-09 15:15:21

1.1K0

发布于 2018-01-09 15:15:21

tensorflow下设置使用某一块GPU（从0开始编号）：

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "1"

多GPU：

num_gpus = 4

for i in range(num_gpus):
    with tf.device('/gpu:%d',%i):
        。。。

只是用cpu的情况

with tf.device("/cpu:0"):

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2017年10月20日，如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体分享计划，欢迎热爱写作的你一起参与！

登录后参与评论

0 条评论

热度