GPU的使用配置

首先，为了明确使用的GPU型号，ubuntu系统下可以利用终端指令查看系统GPU的使用、型号情况。

sudo nvidia-smi

输出的结果如下：

也可以在代码中通过tensorflow_backend获取到可用的GPU信息。

from keras import backend as K
print(K.tensorflow_backend._get_available_gpus())

输出结果如下：

接下来，在代码中通过CUDA_VISIBLE_DEVICES来选定GPU：

import tensorflow as tf
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

将编号选择成对应的GPU编号即可选定该GPU训练模型。

若要选定多块GPU，将相应的GPU编号用逗号隔开，即可。

os.environ["CUDA_VISIBLE_DEVICES"] = "0，1"

配置GPU内存的占用率

在上述已经选定完成GPU的使用配置后，可以选择训练模型时对GPU内存的占有率，防止为了训练模型而影响服务器的其他任务正常使用。

gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
for gpu in gpus:tf.config.experimental.per_process_gpu_memory_fraction = 0.7   #可调整GPU内存的占有率

通过上述两个配置，即可实现在GPU训练环境已配置（cuda、cudnn等）的情况下，实现GPU训练Keras2.x的深度学习模型。

本文仅作为学习笔记交流分享用。由于本人水平有限，文章可能出现纰漏，欢迎大家进行交流指正。谢谢阅读！

利用GPU训练的配置方法（Keras）相关推荐

利用GPU训练网络模型
文章方式一方式二本文章中使用的网络模型架构图: GPU训练有两种方式: 方式一使用gpu训练只要找到:网络模型.数据(输入和标注).损失函数再调用.cuda()即可. CPU训练代码: imp ...
【Pytorch神经网络实战案例】01 CIFAR-10数据集：Pytorch使用GPU训练CNN模版-方法①
import torch import torchvision from torch import nn from torch.utils.tensorboard import SummaryWrit ...
【Pytorch神经网络实战案例】02 CIFAR-10数据集：Pytorch使用GPU训练CNN模版-方法②
import torch import torchvision from torch import nn from torch.utils.tensorboard import SummaryWrit ...
windows10下 tensorflow gpu版本安装配置方法
最近要用到tensorflow,之前安装了cpu版本的,但训练模型速度实在是慢,打算安装一个GPU版本的tensorflow: 安装环境为:wiindows 10, anaconda python36 ...
利用GPU训练网络时遇到的一些问题
1. OSError: [Error 12] Cannot allocate memory 解决办法: 出现这个错误时我是绝望的...因为我看了别人的解决办法,要加内存条才能解决...但是我不甘心,想 ...
TensorFlow2.4可以在MacBook Pro/Mac Pro上利用GPU做机器学习训练了
田海立@CSDN 2020-11-21 之前MacBook上TensorFlow只能利用CPU做训练,TF2.4开始可以利用GPU做训练了,并且不管是M1的MacBook Pro,还是Intel架构的 ...
CPU和GPU训练耗时比较
网络训练耗时为什么要用GPU训练网络 CPU 训练是在载入图片并进行处理后再开始训练网络,而利用GPU 训练是在CPU 载入图片处理后利用GPU 进行网络训练,GPU 训练速度更快且后面的图片加载不 ...
深度学习入门（三十八）计算性能——多GPU训练
深度学习入门(三十八)计算性能--多GPU训练前言计算性能--多GPU训练课件多GPU并行数据并行VS模型并行数据并行总结教材 1 问题拆分 2 数据并行性 3 简单网络 4 数据同步 ...
使用估算器、tf.keras 和 tf.data 进行多 GPU 训练
文 / Zalando Research 研究科学家 Kashif Rasul 来源 | TensorFlow 公众号与大多数 AI 研究部门一样,Zalando Research 也意识到了对创意 ...

利用GPU训练的配置方法（Keras）

利用GPU训练的配置方法（Keras）

GPU的使用配置

配置GPU内存的占用率

利用GPU训练的配置方法（Keras）相关推荐

最新文章

热门文章