甜美钢笔

文章
6
资源
0
加入时间
3年12月3天

python os.environ gpu_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 #导入已经写好的函数模型,例如VGG19if G <= 1:...