记录Ubuntu18.04安装Keras的过程,以TensorFlow-GPU为后端

这里以从开辟一个新的conda环境开始。

1、conda create -n py36-keras python=3.6

2、source activate py36-keras

3、先安装TensorFlow-GPU,并指定版本,这样conda会自己知道对应的GPU加速用到的cuda和cudnn,并自动安装。命令:conda install tensorflow-gpu=1.8.0,这里我指定的是1.8版本,自己一直都是在用这个版本,所以选择了它。遇到需要输入y/n的时候,输入y就行了。

4、安装完成TensorFlow-GPU,这个时候你可以用conda list查看一下。此时TensorFlow-GPU、cudatoolkit、cudnn已经安装完成,并且自动匹配了对应的版本,避免自己去查找版本对应的问题了。见下图:

5、conda install keras

6、大功告成。

发布了141 篇原创文章 · 获赞 54 · 访问量 15万+

猜你喜欢

转载自blog.csdn.net/iamjingong/article/details/98057965