一、电脑的基本信息
显卡:NVIDIA GeForce GTX 1070 Ti
系统:Win 10
二、安装的软件信息
1. Geforce Game Ready Driver 版本:417.22
2. Python 版本:3.6.3
3. Cuda 版本:9.1.85
4. Cudnn 版本: 7.0.5
5. tensorflow-gpu wheel 版本:1.6.0
三、tensorflow-gpu 1.6 支持CUDA9.1的wheel下载地址:
https://github.com/fo40225/tensorflow-windows-wheel/tree/master/1.6.0/py36/GPU/cuda91cudnn71avx2
四、windows上编译tensorflow不同特性的步骤:
https://github.com/fo40225/tensorflow-windows-wheel/issues/3
五、测试是否安装成功
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())