我的环境:
win10, 1080ti, cuda=8.0, cudnn=6.0, keras=2.1.2, tensorflow-gpu=1.3
这几天弄这个问题搞得头都大了,重装过显卡驱动,cuda, cudnn各种版本都试了,网上也找不到有效的解决办法。
最后因为在别的win10电脑上程序可以跑,详细比较了下两个电脑的环境,最后终于找到原因了!!!
python 版本从3.5.5改成3.5.2就没问题了!!!
搞深度学习版本问题真的坑,还是pytorch大法好!!!
我的环境:
win10, 1080ti, cuda=8.0, cudnn=6.0, keras=2.1.2, tensorflow-gpu=1.3
这几天弄这个问题搞得头都大了,重装过显卡驱动,cuda, cudnn各种版本都试了,网上也找不到有效的解决办法。
最后因为在别的win10电脑上程序可以跑,详细比较了下两个电脑的环境,最后终于找到原因了!!!
python 版本从3.5.5改成3.5.2就没问题了!!!
搞深度学习版本问题真的坑,还是pytorch大法好!!!