tensorflow win10 系统下安装

安装gpu版本 Step1 安装CUDA8.0

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

Step2 .0

去官网找到对应版本的cuDNN下载,简单注册后即可进入下载页面

@ >

选择cuDNN v6.0(201年4月27日7),供CUDA8.0下载。下载完成后会得到一个压缩包,解压后复制将压缩包中的bin、lib和三个文件夹粘贴到CUDA的安装目录8.0.

如果默认安装了CUDA8.0,那么安装目录应该是C: Files GPU CUDAv8.0。

然后 C: Files GPU CUDAv8.0\CUPTI 与 C: Files GPU CUDAv8.0libx64 到环境变量PATH .

图片[1]-tensorflow  win10 系统下安装-唐朝资源网

Step3 安装 .6

进入下载3.6版本,自带.6。

Step4安装

进入Step3安装后,进入交互环境,输入pip3 — -gpu回车即可下载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)

查看输出结果

即成功。

© 版权声明
THE END
喜欢就支持一下吧
点赞286 分享
评论 抢沙发
头像
欢迎您留下宝贵的见解!
提交
头像

昵称

取消
昵称表情代码图片