1、keas如果搭配Tensorflow1.x用,就一定要用先安装tensorflow-gpu、然后安装keras-gpu。
在这里插入代码片
conda install tensorflow-gpu==1.15
conda install keras-gpu==2.3.1
不能安装keras,而是要安装keras-gpu。因为Tensorflow1.x是区分CPU版本和GPU版本的。
2、keras如果搭配Tensorflow2.x用,则keras不必区分CPU版本和GPU版本。
具体keras和tensorflow的搭配表,见下图。