Keras/Tensorflow选择GPU/CPU运行

首先,导入os,再按照PCI_BUS_ID顺序,从0开始排列GPU,

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

然后就可以选择用哪一个或者那几个GPU运行:

os.environ[”zCUDA_VISIBLE_DEVICES”] = "0"

用0号GPU,即‘/gpu:0‘运行;

os.environ[“CUDA_VISIBLE_DEVICES”] = “0,1”

用0号和1号设备,即‘/gpu:0‘、‘/gpu:1‘运行;

os.environ[“CUDA_VISIBLE_DEVICES”] = “7”

用7号设备,即‘/gpu:7‘运行,但是‘/gpu:7‘根本不存在,于是就只能用CPU运行了。

原文地址:https://www.cnblogs.com/mstk/p/10202205.html

时间: 2024-11-07 05:56:41

Keras/Tensorflow选择GPU/CPU运行的相关文章

TensorFlow指定GPU/CPU进行训练和输出devices信息

TensorFlow指定GPU/CPU进行训练和输出devices信息 1.在tensorflow代码中指定GPU/CPU进行训练 with tf.device('/gpu:0'): .... with tf.device('/gpu:1'): ... with tf.device('/cpu:0'): ... 2.输出devices的信息 在指定devices的时候往往不知道具体的设备信息,这时可用下面的代码查看对应的信息 进入Python环境 from tensorflow.python.c

Win10 TensorFlow(gpu)安装详解

Win10 TensorFlow(gpu)安装详解 写在前面:TensorFlow是谷歌基于DistBelief进行研发的第二代人工智能学习系统,其命名来源于本身的运行原理.Tensor(张量)意味着N维数组,Flow(流)意味着基于数据流图的计算,TensorFlow为张量从图象的一端流动到另一端计算过程.TensorFlow是将复杂的数据结构传输至人工智能神经网中进行分析和处理过程的系统.从去年十一月开源至今一年多一点的时间里,该项目已经收获了40000+的star和18000+的fork,

[AI开发]centOS7.5上基于keras/tensorflow深度学习环境搭建

这篇文章详细介绍在centOS7.5上搭建基于keras/tensorflow的深度学习环境,该环境可用于实际生产.本人现在非常熟练linux(Ubuntu/centOS/openSUSE).windows上该环境的搭建 :) 前面三篇博客代码实现均基于该环境(开发或者测试过): [AI开发]Python+Tensorflow打造自己的计算机视觉API服务 [AI开发]基于深度学习的视频多目标跟踪实现 [AI开发]视频多目标跟踪高级版 运行环境 1) centOS 7.5 ,不要安装GUI桌面:

linux 安装tensorflow(gpu版本)

一.安装cuda 具体安装过程见我的另一篇博客,ubuntu16.04下安装配置深度学习环境 二.安装tensorflow 1.具体安装过程官网其实写的比较详细,总结一下的话可以分为两种:安装release版本和源码编译安装.因为源码编译安装比较繁琐,且需要安装谷歌自己的编译器bazel,所以我选择安装编译好的. 2.我写这篇博客的时候tensorflow更新到了1.4.0,安装编译好的一定看版本,因为每个版本依赖的底层库是不一样的. 1.4.0版本安装之前需要安装CUDA-8,cuDNN v6

【转】多核CPU运行模式

多核CPU运行模式主要有以下三种: •非对称多处理(Asymmetric multiprocessing,AMP)——每个CPU内核运行一个独立的操作系统或同一操作系统的独立实例(instantiation). •对称多处理(Symmetric multiprocessing,SMP)——一个操作系统的实例可以同时管理所有CPU内核,且应用并不绑定某一个内核. •混合多处理(Bound multiprocessing,BMP)——一个操作系统的实例可以同时管理所有CPU内核,但每个应用被锁定于某

玩转CPU运行曲线

Leaf 是不是从来没有想过看看cpu运行曲线啊骚年?顶多也就仅仅是看看cpu利用率,吓自己一跳后感觉关闭几个不该打开的程序~ 然而问题来了,微软公司要让你绘制cpu运行曲线啊!!不仅是固定的直线,还要绝对值函数,还有正弦!!我的天这游戏还能玩? Require 写一个程序,让用户来决定Windows任务管理器(Task Manager)的CPU占用率.程序越简越好,计算机语言不限.例如,可以实现下面三种情况: CPU的占用率固定在50%,为一条直线 CPI的占用率为一条直线,但是具体占用率由命

Python学习笔记——进阶篇【第八周】———CPU运行原理与多线程

CPU运行原理与多线程 什么是线程(thread)? 线程是操作系统能够进行运算调度的最小单位.它被包含在进程之中,是进程中的实际运作单位.一条线程指的是进程中一个单一顺序的控制流,一个进程中可以并发多个线程,每条线程并行执行不同的任务 进程是容器,线程是真正执行的任务单元

Chapter 1 Securing Your Server and Network(1):选择SQL Server运行账号

原文:Chapter 1 Securing Your Server and Network(1):选择SQL Server运行账号 原文出处:http://blog.csdn.net/dba_huangzj/article/details/37924127  ,专题目录:http://blog.csdn.net/dba_huangzj/article/details/37906349 未经作者同意,任何人不得以"原创"形式发布,也不得已用于商业用途,本人不负责任何法律责任. 前言: S

限制TensorFlow只在CPU上运行的方法

我们知道安装tensorflow可以安装两个版本的,有cpu和gpu的,如果同时安装的时候会默认进行用gpu进行,那如歌更改为使用cpu呢 在import tensorflow/keras 之前,加入如下代码 import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "-1" 原文地址:https: