在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使用递归算法进行数值计算的相关操作技巧,需要的朋友可以参考下2018-07-07
这篇文章主要介绍了Django 限制访问频率的思路详解，本文给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下2019-12-12
Python+OpenCV人脸识别签到考勤系统实现(附demo)
本文主要介绍了Python+OpenCV人脸识别签到考勤系统实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-04-04
这篇文章主要为大家介绍了Java开发者快速进修Python指南之简易跳表示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-12-12
Python是一种广泛使用的编程语言,非常适合处理各种任务,包括批量解压缩文件,本文主要介绍了Python实现批量压缩解压文件,具有一定的参考价值,感兴趣的可以了解一下2023-09-09
Mac系统中Anaconda环境配置Python json库的方法详解
这篇文章主要为大家介绍了如何在Mac电脑的Anaconda环境中,配置Python语言中,用以编码、解码、处理JSON数据的json库,需要的小伙伴可以参考下2023-08-08
这篇文章主要介绍了python爬取有道词典的方法，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2020-12-12
这篇文章主要介绍了opencv 图像加法与图像融合的实现代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-07-07
Python实现K-means聚类算法并可视化生成动图步骤详解
这篇文章主要介绍了Python实现K-means聚类算法并可视化生成动图,本文分步骤给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-05-05
这篇文章主要介绍了Pycharm IDE的安装和使用,本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-04-04
最新评论