Keras中多GPU或指定GPU使用的介绍
5星
- 浏览量: 0
- 大小:None
- 文件类型：None
简介：
本文介绍了在Keras深度学习框架中如何配置和使用多GPU或多张GPU卡中的特定一张进行模型训练的方法。
在Keras的新版本中加入了多GPU并行使用的函数。下面的程序段可以实现一个或多个GPU加速：
注意：使用多GPU加速时，需要确保Keras版本为2.0.9及以上。
```python
from keras.utils.training_utils import multi_gpu_model
import VGG19 # 导入已经写好的VGG19模型函数
if G <= 1:
print(训练一个GPU...)
else:
model = VGG19()
```
如果G小于等于1，则程序会使用单个GPU进行训练；否则，将利用多GPU加速。
全部评论 (0)
还没有任何评论哟~
