码迷,mamicode.com
首页 > Windows程序 > 详细

tensorflow win10 系统下安装

时间:2017-12-30 22:38:48      阅读:332      评论:0      收藏:0      [点我收藏+]

标签:x64   结果   uda   python3   pat   atm   简单的   页面   imp   

安装tensorflow gpu版本

Step1 安装CUDA8.0

进入这个云盘地址下载,密码5aoc 进行CUDA8.0下载。下载完成后解压,打开exe文件直接按照默认进行安装,安装步骤比较繁琐,时间消耗较长,需耐心等待。

Step2 安装cuDNN6.0

进入英伟达官网 找到对应版本的cuDNN进行下载,进行简单的注册后,可以进入下载页面
技术分享图片
选择Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0进行下载。下载完毕后,得到一个压缩包,解压后将压缩包中bin、lib、include三个文件夹复制,粘贴到CUDA8.0的安装目录下。
如果是按默认方式安装的CUDA8.0,那么安装目录应该为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 。
之后将C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\extras\CUPTI\libx64 与C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64添加到环境变量PATH中。
技术分享图片
技术分享图片
技术分享图片

Step3 安装Anaconda3.6

进入Anaconda 下载3.6版本,其中自带python3.6。

Step4 安装tensorflow

进入Step3安装完成的Anaconda Prompt, 输入python 进入python交互环境后,输入pip3 install --upgrade tensorflow-gpu 回车,即进行tensorflow gpu版本的下载。
等待下载完成,用样例测试

import tensorflow as tf
a = tf.random_normal((100, 100))
b = tf.random_normal((100, 500))
c = tf.matmul(a, b)
sess = tf.InteractiveSession()
sess.run(c)

查看输出结果
技术分享图片
即为成功。

tensorflow win10 系统下安装

标签:x64   结果   uda   python3   pat   atm   简单的   页面   imp   

原文地址:https://www.cnblogs.com/zddd/p/8151268.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!