训练的时候当然用gpu,速度快呀。我想用cpu版的tensorflow跑一下,结果报错,这个错误不太容易看懂。大概意思是没找到一些节点。后来发现原因,用gpu和cpu保存的pb模型不太一样,但是checkpoints文件是通用的。自己的小本本,之前预装有的pycharm+win10+anaconda3+python3的环境2019/3/24重新安装发现:目前CUDA10.1安装不了tensorflow1.13,把CUDA改为10.0即可如果刚入坑,建议先用tensorflw学会先跑几个demo,等什么时候接受不了cpu这乌龟般的速度之时,就要开始尝试让gpu来跑了。cpu跑tensorflow只需要在anaconda3下载。安装cpu跑的tensorflow:我的小本本目前已经是gpu版本,cpu版本下红圈里那个版本就好了!安装好了后直接在python命令中输入如果不报错说明调用成功。
暂无评论