1.安装gpu版本的keras
conda install keras-gpu
注意:要在自己的anaconda环境下安装
2.使用keras里的可视化模型时,出现Failed to import pydot. You must install pydot and graphviz for pydotprint to work 的报错时,解决办法:
pip install pydot-ng
还会出现问题的话再输入:
sudo apt-get install graphviz
问题解决。