指定tensorflow运行的GPU或CPU设备

  • 如果 TensorFlow 指令中兼有 CPU 和 GPU 实现,当该指令分配到设备时,GPU 设备有优先权。

  • 如果你的系统里有多个 GPU, 那么 ID 最小的 GPU 会默认使用。

当我们要指定tensorflow运行的GPU或CPU设备时,可以使用tf.device()命令

首先查看可用运算设备(CPU,GPU)

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

得到类似以下的输出结果

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 13177083330855175469
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 10968950375
locality {bus_id: 1
}
incarnation: 6161624703599064583
physical_device_desc: "device: 0, name: GeForce GTX 1080 Ti, pci bus id: 0000:00:08.0, compute capability: 6.1"
]

name即是对应设备名称,一般来说(以各自实际情况为准,每个人的情况可能不同)

"/cpu:0":机器的 CPU。
"/device:GPU:0":机器的 GPU(如果有一个)。
"/device:GPU:1":机器的第二个 GPU(以此类推)。

使用命令tf.device()进行指定

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '/device:GPU:0'

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/483892.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

清华大学刘云浩教授——人工智能打开了潘多拉的盒子吗?

来源:计算思维和人工智能2020年8月的第一周,清华大学暑期学校在荷塘雨课堂上“云开学”。开学典礼上,清华大学软件学院刘云浩教授作了一场主题为《What is the role of AI Tomorrow?——人工智能打开了潘多拉的盒子吗?》的讲座。…

【重点!!!】【堆】215.数组中的第K个最大元素

题目 法1:小根堆 最大的K个元素 > 小根堆(类似上窄下宽的梯形) 最小的K个元素 > 大根堆(类似倒三角形) 必须掌握!!! class Solution {public int findKthLargest(int[] nu…

Spring 系列之(1)Spring-framework 调试环境搭建

调试环境: JDK1.8 Gradle IDEA Git 一、开始代码 clone 将 spring-framework 项目 fork 到自己的代码仓,方便自己加注释。下载好 Gradle ,用 idea 打开项目,进行初步构建。 参考文档:官方文档 二、预编译 spring-oxm&a…

计算机网络知识归纳

计算机网络 第一章 概述 1.1 计算机网络在信息时代中的作用 互联网的两个重要基本特点:1.连通性、2.共享 “资源”指的是网络中所有的网络软件、硬件、数据和通信等等资源 硬件资源、软件资源、数据资源、信道资源(信号传输的媒介) “共享…

PyTorch 之 requires_grad,requires_grad_(),grad_fn

x.grad_fn和x.requires_grad为x的属性x.grad_fn:积分方法名,默认为Nonex.requires_grad:是否积分的属性,默认为Falsex.requires_grad_():设置积分的方法,设置之后requires_grad为True """T…

flutter 获取设备屏幕大小

import dart:ui; var s window.physicalSize;print(s);转载于:https://www.cnblogs.com/pythonClub/p/10567811.html

Linux基础入门(一)

Linux为何物 Linux就是一个操作系统,就像你已经已经知道的window、macOS系统一样,至于操作系统是什么,就不用过多解释了 Linux,全称GNU/Linux,是一种免费使用和自由传播的类UNIX操作系统,其内核由林纳斯本…

利用群体智慧,进行大规模的事实审核

来源:混沌巡洋舰为了应对虚假信息,社交媒体网络和新闻机构经常雇佣事实核查员来区分真假。但事实核查人员只能评估网上流传的一小部分故事。麻省理工学院研究人员的一项新研究提出了另一种方法: 研究指出来自普通读者群体的众包准确性判断,实…

torch.randn与torch.rand的区别

randn torch.randn(*sizes, outNone) → Tensor返回一个包含了从标准正态分布中抽取的一组随机数的张量 size:张量的形状 out:结果张量 rand torch.rand(*sizes, outNone) → Tensor[0,1)之间的均匀分布

xsi:schemaLocation=http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd报...

https://blog.csdn.net/qq_36611526/article/details/79067159 今天遇到个问题 文件内引入某个资源 pom.xml头部http://maven.apache.org/xsd/maven-4.0.0.xsd"报错。 后来各种百度发现是包的问题: 1、找到maven的包的位置:window->preferences-…

解决git关联的的文件颜色、小图标

按WinR键打开运行对话框,输入 regedit.exe ,准备修改注册表找到 HKEY_LOCAL_MACHINE\Software\Microsoft\Windows\CurrentVersion\Explorer;新建一个键名称为 “Max Cached Icons” 数据设置为 “2000”;重启一下电脑,…

Cerebras发布全球首个人类大脑规模的AI解决方案

CEREBRAS来源:IEEE电气电子工程师Cerebras Systems公司的CS-2人工智能训练计算机包含了世界上最大的单芯片,该公司透露,在计算机上增加新的存储系统可以提高神经网络的规模,它可以扩大至现有最大模型的100倍,可支持超过…

《动手学深度学习 PyTorch版》学习笔记(一):数据操作

一、数据操作 在PyTorch中,torch.Tensor是存储和变换数据的主要工具。 "tensor"这个单词一般可译作“张量”,张量可以看作是一个多维数组。 标量可以看作是0维张量,向量可以看作1维张量,矩阵可以看作是二维张量。1、t…

计算机系统结构 期末复习

一.名词解释 虚拟机:指通过软件模拟具有完整硬件系统功能的,运行在一个完全隔离环境中的完整计算机系统系统加速比:同一个任务在系统改进前花费总时间和在系统改进后花费总时间的比率Amdahl定律:计算计算机系统中某个…

Redis数据持久化、数据备份、数据的故障恢复

1.redis持久化的意义----redis故障恢复   在实际的生产环境中,很可能会遇到redis突然挂掉的情况,比如redis的进程死掉了、电缆被施工队挖了(支付宝例子)等等,总之一定会遇到各种奇葩的现象导致redis死掉,…

机器学习如何彻底改变游戏中的物理模拟

来源:AI科技评论编译:琰琰编辑:王晔量子力学奠基者之一、英国理论物理学家保罗狄拉克(Paul Dirac)在1929年说过:“大部分物理和化学所需要的数学理论的定律都是已知的,但这些定律的方程太复杂无…

《动手学深度学习 PyTorch版》学习笔记(二):自动求梯度

一、自动求梯度 1、requires_grad_(), detach(), torch.no_grad()的区别 参考博客:https://www.jianshu.com/p/ff74ccae25f3 2、.grad_fn 每个Tensor都有一个.grad_fn属性,该属性即创建该Tensor的Function, 就是说该Tensor是不是通过某些运算得到的&am…

学习计划1

今天开始着手做热词的项目,但是对于数据的处理仍然没有好的办法,网上最多的方法便是用C#操作,但是在尝试之后仍然没有进展转载于:https://www.cnblogs.com/ljysy/p/10574018.html

线性表——顺序表

1 算法思想:顺序遍历整个链表,找出最小值元素,用一个min存储这个值,同时将最后一个元素填补此值#define Maxsize 100 typedef Elemtype int typedef struct{Elemtype data[Maxsize];int length; }bool delmin(sqlist &L,Elem…

重新认识区块链:1550余个应用案例带来的启示

来源:资本实验室,本文摘自《2021全球区块链应用市场报告》当我们谈论区块链的时候,但凡对区块链有所了解的人都能够就相关主题或多或少地表达出自己的一些见解。例如:从技术体系上看,区块链是分布式数据存储、点对点传…