您的位置:首页 > 其它

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

2017-08-15 13:38 591 查看
安装python3.5, Tensorflow目前支持python3.5 和3.6版本,

安装可参考python成功配置scikit-learn,含试错历程

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

安装 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时会出一点问题

通过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填坑笔记
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  tensorflow