win10+tensorflow-gpu(1.12.0)+cuda9.0+cudnn成功配置多台机器

舍友买了几台新笔记本,我们装tensorflow玩玩,实测1060MQ,1060,甚至960显卡都ok
第一步:下载安装anaconda3
第二步:安装tensorflow-gpu
win10+tensorflow-gpu(1.12.0)+cuda9.0+cudnn成功配置多台机器_第1张图片
打开anaconda prompt,输入命令

  • pip install tensorflow-gpu -i https://pypi.doubanio.com/simple/

第三步:安装cuda9.0
在这里插入图片描述
下载cuda9.0,版本如上图所示,双击安装即可
第四步:安装cudnn
在这里插入图片描述
版本如上图所示
同时打开cudnn和cuda9.0的文件夹,如下图:右边是cudnn,左边是cuda
win10+tensorflow-gpu(1.12.0)+cuda9.0+cudnn成功配置多台机器_第2张图片
将右侧bin里的文件复制到左侧bin里,将右侧include里的文件复制到左侧的include里,将右侧lib里的文件复制到左侧lib里。完成
第五步:打开spyder,import tensorflow as tf,报错,如下图所示:
win10+tensorflow-gpu(1.12.0)+cuda9.0+cudnn成功配置多台机器_第3张图片
打开anaconda prompt
pip uninstall protobuf卸载protobuf
pip install protobuf==3.6.0
问题解决,运行程序时用gpu-z可看到gpu在运转
win10+tensorflow-gpu(1.12.0)+cuda9.0+cudnn成功配置多台机器_第4张图片随便找个程序都能跑
win10+tensorflow-gpu(1.12.0)+cuda9.0+cudnn成功配置多台机器_第5张图片

你可能感兴趣的:(tensorflow)