tensorflowgpu版本的windows安装

首先需要明确需求,即自己需要安装的tensorflow-gpu对应的cuda和cudnn版本。通过该tensorflow官方网站可以查看tensorflow对应的cuda和cudnn版本。
在这里插入图片描述
因为我是安装 tensorflow-gpu 的2.2版本,所以从中选择自己需要的cuda和cudnn版本,从而通过cuda官网进行配置

1 配置CUDA与CUDNN

1.1查看本机的CUDA驱动适配版本

桌面右键打开英伟达控制面板,点击帮助->系统信息->组件

从中可以看到我的cuda版本是11.1,则我可以下载11.1以下版本的cuda

1.2 官网下载CUDA

cuda官网给展现了不同的cuda版本,直接选择对应的cuda进行下载即可
在这里插入图片描述
如上图所示,应该只需要下载10.1中这三个的任意一个就好。然后点击去之后就可以看见下图
在这里插入图片描述
选中windows直接下载,或者将下载链接拖到迅雷进行下载,然后安装即可。安装步骤可以借鉴csdn的一个链接
此时打开终端输入 nvcc -V,应该就显示安装成功了。

1.2 官网下载并安装CUDNN

https://developer.nvidia.com/rdp/cudnn-archive#a-collapse51b
从该网站选择对应的cudnn,然后进行下载,下载的过程中需要注册一下自己的账号,直接注册就好。如果不愿意注册,可以直接将下载的链接复制到迅雷,使用迅雷下载。

下载CUDNN后,复制粘贴cuDNN里面的下面三个文件到CUDA的相应同名文件

cudnn\cuda\bin => CUDA\v10.0\bin
cudnn\cuda\include => CUDA\v10.0\include
cudnn\lib\x64 => CUDA\v10.0\lib\x64
下图是一个示意图
在这里插入图片描述

1.3 TensorFlow安装

打开终端,输入:
conda create -n tensorflow pip python=3.8
自己可以选择虚拟环境的名字和python对应的版本。
然后 输入:
conda activate tensorflow
进入该虚拟环境
然后 输入:
pip install --ignore-installed --upgrade tensorflow_gpu==2.2.0
如果下载速度过慢,请将pip更换为国内的源进行下载。
此时应该就安装成功了

这是自己下载的关于cuda10.1和cudnn7.4.2的版本
链接:https://pan.baidu.com/s/1SzFoyNPLQeizeCZgRTzJ_A
提取码:plh5

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值