Win10系统下Tensorflow-gpu1.6.0版安装踩坑实录(安装环境CUDA9.0+cundnn)

本文详细记录了在Win10系统下安装Tensorflow-gpu1.6.0的全过程,包括匹配的CUDA9.0和cudnn的安装步骤。首先确保CUDA版本与TensorFlow匹配,接着下载并安装CUDA,然后安装cudnn,最后通过conda或pip安装Tensorflow-gpu1.6.0,验证安装成功。
摘要由CSDN通过智能技术生成

首先给大家梳理一下TensorFlow1.6.0版本的安装步骤:

1、安装CUDA9.0

2、安装cudnn

3、安装TensorFlow-gpu1.6.0

Part I 安装CUDA

首先,确定你要安装的CUDA版本一定要与TensorFlow版本对应,否则后面安装TensorFlow会报错。

其他的TensorFlow版本我不清楚,但是1.5.0及1.6.0的TensorFlow与CUDA9.0是能够正确匹配的。

注意!!!一定要是CUDA9.0,多一点都不行!如果你安装了CUDA9.2,最后安装TensorFlow的时候就会报错。

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

CUDA9.0安装步骤:按照提示安装就完事了。

Part II 安装cudnn

很多人会疑问为什么还要装这个,这个东西主要是用来给深度学习做加速的,是较新版本的TensorFlow-gou版的必备环境。

下载地址:https://developer.nvidia.com/cudnn

注意,下载这个包需要注册,按照提示注册好之后,下载就行了。

下面重点来了,如何安装cudnn。

STEP1 解压下载的cudnn文件

下载的cudnn大概长

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值