TensorFlow2.0选择GPU或CPU训练

注意：以下所有程序都要在导入TensorFlow和keras包之前加入才可以

选择CPU：

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

选择GPU：

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '/gpu:0'

查看当前设备使用的是CPU还是GPU：

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

TensorFlow2.0选择GPU或CPU训练相关推荐

在tensorflow2.0环境下使用RandLA-Net训练S3DIS数据集
之前的文章介绍了在tensorflow2.0环境下使用RandLA-Net训练Semantic3D数据集,这里我们记录一下如何在在tensorflow2.0环境下使用RandLA-Net训练S3DIS ...
深度学习之选择GPU或CPU方法
1 TensorFlow设置方法 1.1 指定GPU 法1 import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID ...
TensorFlow2.0学习笔记-3.模型训练
3.模型训练 3.1.Keras版本模型训练 • 构建模型(顺序模型.函数式模型.子类模型) • 模型训练: model.fit() • 模型验证: model.evaluate() • 模型预测: ...
tensorflow2.0 GPU 版本安装测试教程及新特性初探
安装与测试 TensorFlow2.0安装: pip install tensorflow-gpu==2.2.0 -i https://pypi.douban.com/simple/ conda in ...
Ubuntu18.04安装anaconda+tensorflow2.0(GPU)、显卡驱动、搜狗输入法、TIM
笔记本基本配置准备工作 ubuntu18.04的安装显卡驱动安装搜狗输入法的安装 deep-wine下TIM的安装双系统时钟不一致问题的解决 Anaconda安装tensorflow2.0(G ...
用matlab卷积神经网络遇到的问题——gpu和cpu的结果不一样，最后的预测结果也有问题
用官方的帮助文档的程序识别手写数字集test里面的图片以下是代码: clc;clear;close all; digitDatasetPath = fullfile('C:\Users\86151\ ...
pip更新失败_最全Tensorflow2.0 入门教程持续更新
最全Tensorflow 2.0 入门教程持续更新: Doit:最全Tensorflow 2.0 入门教程持续更新zhuanlan.zhihu.com 完整tensorflow2.0教程代码请看ht ...
Tensorflow2.0安装教程（windows系统，CPU和GPU版本）
文章目录一.Anaconda安装二.tensorflow2.0安装 2.1 创建Anaconda环境 2.2 进入TF20环境 2.3 Tensorflow安装三.验证tensorflow安装成 ...
tensorflow2.0的cpu与gpu运行时间对比
文章目录前言一.导入环境二.定义函数三.测试前言这里运用一个自定义大小的矩阵数据计算,来测试gpu与cpu运算时间的对比. 以下为实现方法一.导入环境示例:pandas 是基于NumP ...

TensorFlow2.0选择GPU或CPU训练

TensorFlow2.0选择GPU或CPU训练相关推荐

最新文章

热门文章