keras安装分为两种情形:
第一种
之前已经安装好了tensorflow-GPU版的话,直接运行:
sudo pip3 install keras
第二种
如果没有安装的话,请先安装tensorflow-gpu,安装命令是:
sudo pip3 install tensorflow-gpu==1.5 #这里的版本是因为要对应我的系统里的CUDA9+CUDNNv7
然后安装keras:
sudo pip3 install keras
验证下:
import tensorflow
import keras
如果不报错,就是配置成功了~如果想知道关于CUDA和CUDNN的安装,请参考博客:https://blog.youkuaiyun.com/sinat_23853639/article/details/80990967
里面的教程很清晰,推荐借鉴~