Tensorflow-gpu成功安装步骤

写在开头:再次真诚感谢蔡同学对我的指导与帮助。

本人安装tensorflow-gpu版本,参考了很多其他博客,很多步骤都有点瑕疵,本人在这里对其进行总结。

首先检查自己的电脑gpu是否支持cuda,去官网检查https://developer.nvidia.com/zh-cn/cuda-gpus#compute
我安装的是tf-gpu1.4 cuda8.0 cudnn6

1.安装Anaconda

直接去anaconda官网下载最新版即可,因为最新版本带的spyder编译器进行了升级,本人使用过,非常好用;若需要其他版本的python版本,只需要在anaconda中创建对应的环境即可。https://anaconda.org/

https://www.continuum.io/downloads

2.tensorflow及cuda与cudnn版本

网上自找相关对于的版本即可

3.安装CUDA

直接去对应的官网下载即可。一路安装,若出现This graphics driver could not find compatible graphics hardware…是因为自身电脑的硬件显卡版本高于对应cuda自带的,所以自定义安装,只需勾选cuda即可,后面的两个驱动都不需要安装,若出现no supported version of visual studio问题,需要先安装对应的vs,本人测试的是cuda8对应的是vs2015,安装好后检查如下
在这里插入图片描述

在这里插入图片描述
将下载下来的三个文件夹中的文件复制到对应的cuda文件中。

4.安装tf环境

打开anaconda prompt,使用conda create -n tf_gpu python==XX
创建好之后,使用命令:activate tf_gpu进入环境
pip install tensorflow-gpu,及需要的模块,最后进行测试即可:
保存以下代码为test.py
import tensorflow as tf
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name=‘a’)
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name=‘b’)
c = tf.matmul(a, b)
sess=tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run©)
在你的环境下进入你的test.py的目录下,运行以下代码:python test.py,若出现以下截图证明安装成功。
在这里插入图片描述

  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值