GPU or CPU

[w52191]

请问你用的是GPU还是CPU训练的？ 如果是GPU，能否分享一下你的安装环境，比如CUDA，cudnn版本，TensorFlow和Keras的版本等？我尝试过各种组合都没能把GPU版本跑起来。非常感谢！

[xieck13]

hello，运行环境基于 Python 3.6，其中深度学习环境为 Tensorflow背景的 Keras 2.2.4，非侵入式负荷检测平台 版本 为 nilmtk 0.4.0。 tensorflow版本应该是有某个版本是和keras2.2.4匹配的 ，至于CUDA和cudnn的版本只要与tensorflow的版本匹配就行。 有后续其他问题可以加我微信Xieck13 w52191 ***@***.***> 于2021年5月27日周四 下午4:17写道：

…

请问你用的是GPU还是CPU训练的？ 如果是GPU，能否分享一下你的安装环境，比如CUDA，cudnn版本，TensorFlow和Keras的版本等？我尝试过各种组合都没能把GPU版本跑起来。非常感谢！ — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub <#1>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ANXEJHUNPCZPEIFZYN6GLPTTPX53JANCNFSM45TXAV2A> .

[w52191]

微信已加。
那环境是自己创建的还是用NILMTK-CONTRIB推荐的方法，

conda create -n nilm -c conda-forge -c nilmtk nilmtk-contrib

它推荐的这个方法，默认安装Python3.7，Keras和TensorFlow也是比较高的版本。我现在能够运行除RNN以外的其它模型，但是GRU和RNN没办法用cudnn加速。训练比直接跑CPU还慢。