原本已经安装TensorFlow-gpu
下载安装keras-gpu即可:
打开Anaconda Prompt 输入命令:conda install keras-gpu==2.0.8
然后打开Jupyter Notebook =>Kernel=>Change kernel=>Python [conda root] 即可
检验方法1:需要检验是否使用gpu,可以使用代码程序检验。以我为例,相同的程序由原来的500s缩短到20s,表明成功使用gpu。
检验方法2:查看显卡使用率、温度。详情查看windows如何查看nvidia显卡(GPU)的利用率和温度
常见问题:
- 下载了好多不同版本的keras怎么办?
在Anaconda Prompt中输入:conda list 查看当前已经安装的模块和版本。通过 conda uninstall keras==2.2.0 卸载其余的keras版本(我用的是较低版本,所以卸载了最新版。)
参考:
Windows10中使用Anaconda安装keras-gpu版本【入坑经验总结】