TensorFlow深度学习入门——Win10下安装Tensorflow和GPU加速

时间：2022-06-16 22:47:01

  1. 安装python3.5, Tensorflow目前支持python3.5 和3.6版本，
    安装可参考python成功配置scikit-learn，含试错历程

  2. 安装CUDA Driver(https://developer.nvidia.com/cuda-downloads )

  3. 安装 cuDnn -CUDA for Deep Neural Networks
    (http://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html#download-cuda-software )
    我下载的是cuDNN v7.0 Library for Windows 10
    这其实就是CUDA的深度神经网络支持，下载下来以后解压出来一个cuda文件夹，放到你想放的位置，然后把这个cuda文件夹下的bin目录地址放到PATH里。
    如果不在这里配置环境变量的话，结束安装流程以后，import tensorflow时会出一点问题

  4. 通过pip安装tensorflow

    4.1 下载tensorflow（http://www.lfd.uci.edu/~gohlke/pythonlibs/#scipy ）
    tensorflow_gpu 1.1.0 cp35 cp35m win_amd64.whl
    在命令行 pip install tensorflow_gpu 1.1.0 cp35 cp35m win_amd64.whl
    //pip install –upgrade tensorflow_gpu-1.0.0-cp35-cp35m-win_amd64.whl）
    这个网址下载这个文件很慢，我先采用了4.2中的网址进行下载了tensorflow，后期用到GPU加速的时候，需要的话再来安装这一个。

    4.2下载tensorflow（https://pypi.python.org/pypi/tensorflow/1.1.0rc2）

pip install tensorflow-1.1.0rc2-cp35-cp35m-win_amd64.whl

提示：

Installing collected packages: werkzeug, protobuf, wheel, tensorflow
Running setup.py install for protobuf … done
Successfully installed protobuf-3.3.0 tensorflow-1.1.0rc2 werkzeug-0.12.2 wheel-0.29.0

安装成功！

最后测试一下啊

>>> import tensorflow as pf
>>> hello = pf.constant('Hello world!')
>>> sess = pf.Session()
>>> print (sess.run(hello))
b'Hello world!'

参考安装博文

tensorflow及GPU加速在win10下的安装教程
win10上安装tensorflow注意事项
win10安装TensorFlow填坑笔记

标签：

相关文章