keras 使用单个或多个GPU训练模型(已经不需要自己设置了,会自动检测环境)
问题: keras 使用单个或多个GPU训练模型解决办法:单个GPU:直接设置环境import osimport osos.environ["CUDA_VISIBLE_DEVICES"] = "0" # 指定GPU编号def get_model(): # input_ids = keras.layers.Input(shape=(MAXLEN, ))... #_, x = bert_model([input_ids, attension_mask, token_type_ids])
原创
2020-12-04 20:28:44 ·
1148 阅读 ·
0 评论