Tensorflow-GPU在window环境下安装历程

    Tensorflow是Google在2015年11月9日宣布开源的第二代机器学习系统,支持python和C++,支持CNN、RNN和LSTM等算法,这里python IDE推荐pyChorm,功能强大,安装库也很方便,简单的tensorflow的默认cpu版本可以直接pip安装运行没有问题,但是速度非常慢,所以改为使用tensorflow-gpu,速度能提升很多。

     首先查询电脑是否支持gpu,确定支持后,先下载NVIDIR的CUDA和CUDNN(可以在英伟达官网下载),CUDA是图像处理支持,CUDNN是链接tensorflow和CUDA必要的插件,这里CUDA必须下载8.0版本,9.0版本暂时不支持。这里对版本的对应非常严格,之前下载哈工大的pyltp我就体会到了,简直一个版本的pyltp需要一个版本的visio studio。我尝试了很多版本,最终选择了Tensorflow-gpu 1.14.0rc1 + cuda_8.0.61.2_windows + cudnn-8.0-windows10-x64-v6.0 首先下载cuda,之后默认路径安装后有如下文件就算安装成功了。

      然后下载cudnn,解压后有三个文件夹。

      将cudnn各个文件夹中的文件复制到对应CUDA的相同文件夹中。

      最后修改环境变量,安装CUDN后默认的环境变量是CUDA_PATH = C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,没有显示bin目录和x64目录,单独将其添加到path中。最终还要确认有visio studio c++ 2015 Redistributable - x64环境。没有的话会报错 ‘DLL not found’.

     然后打开命令行输入pip install tensorflow-gpu,等待gpu版本的tensorflow安装完成就OK了,输入 python进入python环境,输入 import tensorflow as tf,没有报错就说明安装成功,如果报错,重启一次。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值