GPU or CPU
w52191 opened this issue · 2 comments
w52191 commented
请问你用的是GPU还是CPU训练的？ 如果是GPU，能否分享一下你的安装环境，比如CUDA，cudnn版本，TensorFlow和Keras的版本等？我尝试过各种组合都没能把GPU版本跑起来。非常感谢！
xieck13 commented
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 commented
微信已加。
那环境是自己创建的还是用NILMTK-CONTRIB推荐的方法，
conda create -n nilm -c conda-forge -c nilmtk nilmtk-contrib
它推荐的这个方法，默认安装Python3.7，Keras和TensorFlow也是比较高的版本。我现在能够运行除RNN以外的其它模型，但是GRU和RNN没办法用cudnn加速。训练比直接跑CPU还慢。