Win10下安装tensorflow(gpu版)

Win10下安装tensorflow(gpu版)

我安装的环境

系统:win10系统
python:3.5
GPU:GeForce GTX1660Ti
CUDA:cuda_9.0.176_win10
CUDNN:cudnn-9.0-windows10-x64-v7

1.安装anaconda

我安装的时anaconda版本是4.2.0,安装包在官网下载,anaconda下载,安装包下载完成后安装,安装过程很简单,一直next即可。

2.安装vs2017(vs2015也可以)

该步很重要,一定要确保安装了vs,若是没有安装vs,直接安装CUDA会导致安装失败,出现很多问题。
vs2017下载

3.安装cuda9.0

CUDA安装过程中会出现各种各样的坑,我也是安装了好多次才成功的。
该部分安装失败可参考博(https://blog.csdn.net/zzpong/article/details/80282814)和(https://blog.csdn.net/xhbspark/article/details/89929310)

4.CUDA9.0安装后测试

1)打开cmd,输入两次cd… ,进入CUDA9.0安装路径下的demo_suite文件夹
在这里插入图片描述
2)运行该文件夹下的两个.exe文件
在这里插入图片描述
在这里插入图片描述
若result=PASS,则表示安装成功,否则安装失败,需要重新安装

5.下载cudnn

这个步骤比较简单,只需要将解压文件中的文件放在CUDA9.0安装路径中的相应文件夹中即可。

6.安装tensorflow-gpu

在下面表中找出对应的tensorflow版本
在这里插入图片描述
使用如下命令安装tensorflow-gpu:
pip install --ignore-installed --upgrade tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

7.测试安装是否成功

运行如下程序测试:

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

若运行后结果如红色框中所示,则表示安装成功
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值