win10安装tensorflow gpu

step 1:

安装anaconda

官网下载有点慢

推荐使用清华镜像+迅雷下载,下载链接如下

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.1.0-Windows-x86_64.exe

可以在anaconda下创建一个专门用于tensorflow的环境

该环境使用python 3.5.x

anaconda prompt下输入命令:

conda install tensorflow-gpu

安装tensorflow即可

step 2:

安装cuda和cudnn

安装cuda的过程中,选择“自定义安装”,仅勾选cuda即可

(如果选择精简安装,安装过程中会重启,重启过后会无法继续安装)

之后将cudnn文件夹中的三个文件复制到cuda的computing toolkit下


之后,需要设置环境变量,在path中需要包含以下路径:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64


step 3:

在anaconda prompt中,激活之前创建的tensorflow环境

进入python

输入import tensorflow as tf 进行测试


没报错就是安装成功啦!



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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值