Windows10-64bit 上的GPU版本Tensorflow安装

本文详述了在Windows10 64位系统上安装GPU版本Tensorflow的步骤,包括安装CUDA V9.0、cuDNN v7.0.4,并解决版本对应问题,最后展示GPU加速训练的效果。
摘要由CSDN通过智能技术生成

Win10 GPU版本Tensorflow安装

最近由于练习主要集中在机器学习处理图片的各种算法部分,CPU版本的“Tensorflow”已经无法满足使用需求,所以进行了GPU版本的“Tensorflow”的安装,以下为一些问题的记录。

0、Note

  • GPU Tensorflow只支持“Nvidia”显卡,所以“AMD”显卡用户无需尝试
  • GPU Tensorflow需要“CADU”和“cuDNN”的支持,出现的问题也就在此,主要是相互间的版本对应及安装后的环境变量的配置问题。
  • GPU Tensorflow使用的版本为“1.7.0”,不同版本需要的“CADU”和“cuDNN”版本不同,可自行查阅相关文档,“1.7.0”当前仅支持V9.0的“CADU”

1、安装CADU

CUDA V9.0下载地址: https://developer.nvidia.com/cuda-90-download-archive

CADU

具体根据系统等等信息进行选择。这里的重点是此文成文时“CUDA”最新版本为V9.1,这里使用的V9.0的原因是因为前述提到将安装的“1.7.0”版本的Tendorflow支持的“CUDA”版本为V9.0,所以请对应安装

安装包下载速度不错,

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值