同时存在keras和keras-gpu,如何调用keras-gpu
时间: 2023-12-16 08:04:31 浏览: 94
tensorflow1.14-gpu+keras环境
你可以在代码中使用以下方式来调用 keras-gpu:
```python
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0" # 调用的GPU编号,如果有多个GPU可以使用逗号隔开,例如:os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2"
import tensorflow as tf
config = tf.compat.v1.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.compat.v1.Session(config=config)
from keras import backend as K
K.tensorflow_backend.set_session(session)
```
这样就可以使用 keras-gpu 进行训练和推理了。
阅读全文