Window安装TensorFlow- GPU环境

本文提供了一种详细的TensorFlow-GPU安装流程,避免了网上误导性的CUDA8.0安装陷阱,适合使用Anaconda3、CUDA-9.0、CuDNN-7.0和Python-3.5环境的用户。文章涵盖了从Python安装、Anaconda3安装、CUDA和CuDNN安装到Tensorflow-GPU安装的全过程,并提供了测试方法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

简述】

  关于Window安装TensorFlow- GPU环境的文章我想网站已经有非常多了,但是为什么还要写这篇文章呢,就是被网上的文章给坑了。由于pip install tensorflow-gpu的时候会默认安装tensorflow的最新版本(目前为1.8)。但是网上的教程估计都是小与1.7的版本,因此所有文章都写着必须安装DUDA8.0,导致小编安装TensorFlow- GPU入了很多坑,最后还是通过看错误信息,才顺利安装上的,因此分享这篇文章,避免小伙伴们入CUDA8.0的坑。

【所需环境】

  • Anaconda3(64bit)
  • CUDA-9.0
  • CuDNN-7.0
  • Python-3.5
  • TensorFlow-gpu1.7版本以上

【Python安装】

  1、下载Python3.5

  2、安装Python3.5(直接安装即可,路径可以改成自己设定的路径)。

  3、安装完成后进入cmd命令行输入python,查看是否安装成功。

【安装Anaconda3(64bit)】

  1、下载Anaconda3(64bit)

  2、安装软件(直接安装即可,不做详细解释)。

【安装CUDA和CuDNN】

  1、下载CUDACuDNN

  2、下载CUDNN时需要注册并填写相关资料;

  3、安装CUDA前,先升级显卡驱动,要不然后面运行程序是报错(注意!!!);

  4、安装CUDA,选择自定义安装,并只选择CUDA其他组件不安装;

  5、安装好CUDA后,解压cuDNN,压缩包里面有三个文件夹,把这三个文件夹复制到你cuda的安装目录下,我的地址是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

       6、安装好后,在系统环境PATH中添加cuda安装路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin

 

 

 

【安装Tensorflow-GPU】

  1、进入cmd,输入pip install tensorflow-gpu==1.8.0;

  2、如果提示需要升级pip,则按照提示命令自行升级;

【测试】

  最后,我们进入jupyter notebook(Anaconda自带的Python IDE,自我感觉挺好用的)输入一段官方文档录入的代码测试一下:

 View Code

  如果不报错,并输出成功,则证明安装成功。

源码获取方式,关注公总号RaoRao1994,查看往期精彩-所有文章,即可获取资源下载链接

更多资源获取,请关注公总号RaoRao1994

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

扰扰1994

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值