终于把tensorflow的gpu版本给跑起来了
碰到的几个比较大的问题 首先是安装pip uninstall tensorflow 把cpu版本的tensorflow给删了
然后pip install tensorflow-gpu 这个时候要注意一下 默认安装的是最新版本的tensorflow-gpu版本 然后现在1.6版本
1.6版本貌似是要用cuda9.0版本的 由于之前用caffe 下载的cuda8.0版本 使得出错 tensorflow包导入不了,很尴尬。OSError: [WinError 126] 找不到指定的模块/Could not find 'cudart64_90.dll'.大概是报这个错,这个错是说没有找到cuda9.0版本。这个时候我去查了一下cuda8.0对应的要是tensorflow-gpu 1.3版本 这个时候就要安装1.3版本的tensorflow-gpu 命令为