win10+cuda+tensorflow+gpu的安装图解
随着时代的发展,人工智能登上了时代的舞台,由谷歌大脑推出的tensorflow是目前学习人工智能的主流框架之一。由于深度学习所要计算的数据量异常庞大,所以要使用GPU进行加速。所以tensfolw有两种安装方式:1.基于GPU的tensflow的安装方式 2.纯tensorflow的安装方式,在安装时,使用 Anaconda编译器内部安装比较简单,具体流程如下:
- 纯tensorflow安装过程如下:
- 创建一个环境变量,anaconda可以管理多个python版本,这里可以创建python3.6 or python3.5
- 点击python3.6环境,具体如下图:
- 之间点击安装即可,安装完成后,选择python3.6环境,点击启动按键,选择 open with python,打开后输入 import tensorflow,没有任何的错误则安装成功
- 基于GPU的tensflow的安装方式
- 首先安装显卡驱动,本人的电脑是win10系统,需要准备的安装包如下:
- 首先安装cuda_8.0.61_win10.exe,具体流程如下:
-
- 路径可以使用默认
- 选择定义
-
-
- 中间省略一些了简单的步骤。至此安装完毕
- 安装cuda_8.0.61.2_windows.exe
-
- 之后一直下一步即可
- cudnn-8.0-windows10-x64-v6.0.zip,解压这个文件,将文件夹里面内容替换掉C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0下的内容。
- 之后配置环境变量如下:
-
- 安装tensorflow-gpu
- 选择python3.5具体如下
-
- 安装完毕后,切记重启电脑。
- 写个测试程序 import tensorflow as tf
- h=tf.constant("hello world!")
- sess=tf.Session()
- print(sess.run(h))
- 输出 “hello world”,至此安装成功