windows下安装GPU版tensorflow并对GPU进行测试

本文详细介绍了在Windows系统中安装Tensorflow-GPU的步骤,包括显卡驱动的安装、CUDA与cuDNN的配置,以及通过pip安装tensorflow-gpu。还提供了检查安装成功的方法和使用python代码进行GPU测试的流程。
摘要由CSDN通过智能技术生成

首先安装显卡驱动

显卡驱动不会对下面的安装产生影响

CUDA、cuDNN以及tensorflow-gpu版本对应关系

在这里插入图片描述

安装CUDA

首先在CUDA官网上下载所需要的CUDA版本:https://developer.nvidia.com/cuda-downloads
在这里插入图片描述
按照提示安装CUDA,记录安装路径,默认路径为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

安装cuDNN

在cuDNN官网下载CUDA版本所对应的cuDNN版本:https://developer.nvidia.com/cudnn (需要登陆,可以用微信登陆)
下载好之后直接解压文件
讲下列文件复制到CUDA相应的位置即可
Copy \cuda\bin\cudnn64_7.dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUD

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值