请选择 进入手机版 | 继续访问电脑版
查看: 1560|回复: 1

tensorflow训练使用GPU和CPU的不同指定方法

[复制链接]

665

主题

1234

帖子

6561

积分

xdtech

Rank: 5Rank: 5

积分
6561
发表于 2020-4-12 09:51:23 | 显示全部楼层 |阅读模式
1.tensorflow如何指定使用CPU跑,并指定使用的CPU个数
cpu_num=10#指定使用的CPU个数
config = tf.ConfigProto(device_count={"CPU": cpu_num},
            inter_op_parallelism_threads = cpu_num,
            intra_op_parallelism_threads = cpu_num,
            log_device_placement=True)
# 开始训练
with tf.Session(config=config) as sess:
    #以下编写自己的代码

2.tensorflow如何指定使用CPU跑(第2种方法)
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"  
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

3.tensorflow如何指定使用的GPU,使其在第k块GPU上跑
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"#指定在第0块GPU上跑


回复

使用道具 举报

665

主题

1234

帖子

6561

积分

xdtech

Rank: 5Rank: 5

积分
6561
 楼主| 发表于 2020-4-12 09:51:33 | 显示全部楼层
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表