keras & tensorflow 列出可用GPU 和 切换CPU & GPU
2024-08-27 00:47:07
列出可用GPU
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices()) from keras import backend as K
K.tensorflow_backend._get_available_gpus()
切换
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
# The GPU id to use, usually either "0" or "1"
os.environ["CUDA_VISIBLE_DEVICES"] = ""
os.environ["CUDA_VISIBLE_DEVICES"] = "-1" 为使用CPU
最新文章
- C学习笔记 知识集锦(二)
- 移动端rem页面详谈
- hdu 4070 福州赛区网络赛J 贪心 ***
- js判断页面点击事件
- linux tricks 之 bitmap分析.
- Java 对象内存分配与回收
- c++操作io常见命令
- c++数组-矩阵的转置
- Javascript高级程序设计学习笔记一
- 评侯捷的<深入浅出MFC>和李久进的<MFC深入浅出>
- LA 3027 Corporative Network 并查集记录点到根的距离
- [实例]JAVA调用微信接口发送图文消息,不用跳到详情页
- 给ASP.NET Core Web发布包做减法
- 新建maven项目错误处理
- Python语法教程总结规范
- 80x86的内存寻址机制
- hbase的wordcount
- 几个方便进行micro frontend 开发的工具&;&;类库
- ASP.NET开发,从二层至三层,至面向对象 (3)
- databus编译:Could not resolve all dependencies for configuration ':databus2-relay:databus2-event-producer-mock:compile
热门文章
- .NET Core跨平台的奥秘[中篇]:复用之殇
- 再谈反向传播(Back Propagation)
- [Swift]LeetCode662. 二叉树最大宽度 | Maximum Width of Binary Tree
- [Swift]LeetCode813. 最大平均值和的分组 | Largest Sum of Averages
- Ubuntu终端常用快捷键(精简)
- Python内置函数(6)——bool
- tensorflow 1.0 学习:十图详解tensorflow数据读取机制
- CentOS 7 源码编译安装 Nginx
- MyBatis 分页插件 PageHelper 使用
- Jenkins 无法捕获构建脚本错误问题