1.查看keras和tf的对应版本
https://docs.floydhub.com/guides/environments/
2.keras如何使用GPU?
https://blog.csdn.net/qq_33182424/article/details/106080243
检查GPU:
from tensorflow.python.client import device_lib print(device_lib.list_local_devices())
检查是否可利用GPU:
from keras import backend as K K.tensorflow_backend._get_available_gpus()
3.判断tf是否可以使用GPU?
https://blog.csdn.net/xiaozisheng2008_/article/details/80558914
import os import tensorflow as tf import numpy as np hello=tf.constant('hhh') sess=tf.Session() print (sess.run(hello))
出现上面这样的就说明可以用了。
然后安装对应版本的keras,之后再使用keras,有如下代码
import os os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"]="0"
就可以使用gpu了。
4.安装keras-contrib
https://blog.csdn.net/qq_31456593/article/details/89388107
pip install git+https://www.github.com/keras-team/keras-contrib.git