您的位置：首页 > 其它

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

2017-10-20 10:14 609 查看

你要的答案或许都在这里：小鹏的博客目录

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"):

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理

标签： tf设置gpu tf设置cpu 多GPU

相关文章推荐

新的分享

章节导航