在tensorflow中设置使用某一块GPU、多GPU、CPU的操作

更新时间：2020年02月07日 16:24:11 作者：MachineLP

今天小编就为大家分享一篇在tensorflow中设置使用某一块GPU、多GPU、CPU的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

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

以上这篇在tensorflow中设置使用某一块GPU、多GPU、CPU的操作就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

深入解析Python中函数的参数与作用域
这篇文章主要介绍了Python中函数的参数与作用域,是Python入门学习中的基础知识,需要的朋友可以参考下
2016-03-03
python对Excel按条件进行内容补充(推荐)
这篇文章主要介绍了python对Excel按条件进行内容补充的相关知识，非常不错，本文通过实例代码给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下
2019-11-11
python使用Qt界面以及逻辑实现方法
这篇文章主要介绍了python使用Qt界面以及逻辑实现方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-07-07
Tortoise-orm信号实现及使用场景源码详解
这篇文章主要为大家介绍了Tortoise-orm信号实现及使用场景源码详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2023-03-03
设置jupyter中DataFrame的显示限制方式
这篇文章主要介绍了设置jupyter中DataFrame的显示限制方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来
2020-04-04
Python使用APScheduler实现定时任务过程解析
这篇文章主要介绍了Python使用APScheduler实现定时任务过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-09-09
pycharm社区版安装node.js插件运行js代码方法
PyCharm可以说是当今最流行的一款Python IDE了,下面这篇文章主要给大家介绍了关于pycharm社区版安装node.js插件运行js代码的相关资料,文中通过代码介绍的非常详细,需要的朋友可以参考下
2023-10-10
OpenCV半小时掌握基本操作之傅里叶变换
这篇文章主要介绍了OpenCV基本操作之傅里叶变换，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2021-09-09
关于PyCharm安装后修改路径名称使其可重新打开的问题
这篇文章主要介绍了关于PyCharm安装后修改路径名称使其可重新打开的问题,本文通过图文实例相结合给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2020-10-10
Windows和Linux下Python输出彩色文字的方法教程
这篇文章主要介绍了在Windows和Linux中Python输出彩色文字的方法，通过设置彩色文字给大家更醒目的效果，文中给出了详细的介绍和示例代码，需要的朋友可以参考借鉴，下面来一起看看吧。
2017-05-05

在tensorflow中设置使用某一块GPU、多GPU、CPU的操作

相关文章

最新评论

大家感兴趣的内容

最近更新的内容

常用在线小工具