Windows 10 安装 Tensorflow GPU版本

Windows 10 安装 Tensorflow GPU版本

PC配置： 小米笔记本

显卡：NVIDIA GeForce MX150 显存2G

内存：8GB

OS：windows 10

 

1.下载Python 3.5

据说Windows10 只可以安装 Python 3.5 版本

登录 python.org 下载3.5的，安装...

下载具体细节不表，博主下载的是exe的

2.进入cmd

输入python --version看看python的版本对不对

博主用的是3.5.3

这里可以参考 tensorflow官网（www.tensorflow.org）

不过好像得才可以

在黑框框里输入：

pip3 install --upgrade tensorflow-gpu

（博主是独立显卡，所以选择gpu的）

等待一切安装完毕

3.还需要安装cuda和cudnn

先安装cuda，然后

将cudnn 下载完的 cudnn64_X.dll 放在cuda的bin文件夹中

保证可以在环境变量里直接用到

有坑预警，如果后期报错，请将cudnn64_6.dll 重命名为cudnn64_5.dll

这样报错就解除了

4.开始第一个程序吧

在python里

>>> import tensorflow as tf

>>> hello = tf.constant(‘Hello,Tensorflow’)

>>> sess=tf.Session()

>>>print(sess.run(hello))

然后就会显示

Hello,Tensorflow

啦

恭喜你，已经做好准备工作啦！

如果你在安装过程中遇到了很多坑，多搜索，看看Stack Overflow。

欢迎入坑