安装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 .
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)
查看输出结果
即成功。
© 版权声明
本站下载的源码均来自公开网络收集转发二次开发而来,
若侵犯了您的合法权益,请来信通知我们1413333033@qq.com,
我们会及时删除,给您带来的不便,我们深表歉意。
下载用户仅供学习交流,若使用商业用途,请购买正版授权,否则产生的一切后果将由下载用户自行承担,访问及下载者下载默认同意本站声明的免责申明,请合理使用切勿商用。
THE END
暂无评论内容