没有合适的资源?快使用搜索试试~ 我知道了~
1. keras新版本中加入多GPU并行使用的函数 下面程序段即可实现一个或多个GPU加速: 注意:使用多GPU加速时,Keras版本必须是Keras2.0.9以上版本 from keras.utils.training_utils import multi_gpu_model #导入keras多GPU函数 import VGG19 #导入已经写好的函数模型,例如VGG19 if G <= 1: print("[INFO] training with 1 GPU...") model = VGG19() # otherwise, we are compiling using m
资源推荐
资源详情
资源评论
keras实现多实现多GPU或指定或指定GPU的使用介绍的使用介绍
1. keras新版本中加入多新版本中加入多GPU并行使用的函数并行使用的函数
下面程序段即可实现一个或多个GPU加速:
注意:使用多GPU加速时,Keras版本必须是Keras2.0.9以上版本
from keras.utils.training_utils import multi_gpu_model #导入keras多GPU函数
import VGG19 #导入已经写好的函数模型,例如VGG19
if G <= 1:
print("[INFO] training with 1 GPU...")
model = VGG19()
# otherwise, we are compiling using multiple GPUs
else:
print("[INFO] training with {} GPUs...".format(G))
# we'll store a copy of the model on *every* GPU and then combine
# the results from the gradient updates on the CPU
with tf.device("/cpu:0"):
# initialize the model
model1 = VGG19()
# make the model parallel(if you have more than 2 GPU)
model = multi_gpu_model(model1, gpus=G)
2.指定使用某个指定使用某个GPU
首先在终端查看主机中GPU编号:
watch -n -9 nvidia-smi
查询结果如下所示:
显示主机中只有一块GPU,编号为0
2.1 下面方法是直接在终端运行时加入相关语句实现指定下面方法是直接在终端运行时加入相关语句实现指定GPU的使用的使用
export CUDA_VISIBLE_DEVICES=0 python test.py
# 表示运行test.py文件时,使用编号为0的GPU卡
export CUDA_VISIBLE_DEVICES=0,2 python test.py
# 表示运行test.py文件时,使用编号为0和2的GPU卡
2.2 下面方法是在下面方法是在Python程序中添加程序中添加
import os
# 使用第一张与第三张GPU卡
os.environ["CUDA_VISIBLE_DEVICES"] = "0, 2"
补充知识:补充知识:踩坑记踩坑记—-large batch_size cause low var_acc
资源评论
weixin_38640443
- 粉丝: 10
- 资源: 887
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- 老照片修复教程(带资料),再也不用去照相馆修复了!-教程网盘链接提取码下载 .txt
- STM32单片机FPGA毕设电路原理论文报告一种建立单片机应用系统菜单的新方法
- 诊所医院超常规运营管理思维,课程+直播+社群+连麦+实操-教程网盘链接提取码下载 .txt
- STM32单片机FPGA毕设电路原理论文报告一种简单方法实现基于STC89C52RC单片机的频率计
- STM32单片机FPGA毕设电路原理论文报告一种基于低功耗单片机的抗干扰电源
- Linux知识体系复习
- STM32单片机FPGA毕设电路原理论文报告一种基于等效采样技术的高速数据采集系统
- 东华深度学习课后习题全部
- STM32单片机FPGA毕设电路原理论文报告一种基于单片机系统的无线遥控技术
- STM32单片机FPGA毕设电路原理论文报告一种基于单片机及CPLD测试平台的设计与实现
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功