Step1.安装cuda_8.0.61_win10.exe(TensorFlow要求cuda8.0)
安装过程时间稍长,中间会出现屏幕闪烁的情况,正常,耐心等待(天气太热,吃瓜中。。。)
安装完了在控制面板里面是这个样子。
Step2:下载cudnn1.5(适配TensorFlow1.2)并添加环境变量(这里不再赘述下载过程,因为需要注册账户)
Step3:安装Anaconda3-4.2.0(Python3.5)
之前查到资料说是Windows下TensorFlow只支持到Python3.5,但是感觉应该是和TensorFlow的版本有关,目前TensorFlow已经出到1.9,应该可以适配最新的Python3.7(个人猜测),但是版本太新总是会有bug,所以选择安装TensorFlow1.2+Python3.5
Step4:安装TensorFlow和TensorFlow-GPU-1.2.0
1.安装TensorFlow
conda create -n tensorflow python=3.5
2.安装TensorFlow-GPU(使用镜像安装)
activate tensorflow
pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.2.0-cp35-cp35m-win_amd64.whl
Step5:测试GPU是否安装成功
1.nvcc -V
2.activate tensorflow
python
import tensorflow as tf
hello=tf.constat("hello tensorflow")
sess=tf.Session()
总体来说,安装情况是这样的,给大家做个参考。如有问题欢迎讨论。