cuda11.2+cudnn8.1+tensorflow2.5.0环境配置

cuda11.2+cudnn8.1+tensorflow2.5.0环境配置

首先查看自己电脑是否为英伟达显卡,如果是则往下进行,查看tensorflow与cuda,cudnn对应版本,网址这里在这里插入图片描述
1. 下载cuda
网址这里,找历史版本11.2,因为我电脑是3060TI,所以选择了11版本的cuda,cuda版本要低等于cuda驱动的版本。
在这里插入图片描述
然后双击安装,点击自定义安装,如下图
在这里插入图片描述
因为驱动版本与cuda本身版本不一致,所以将下图中勾选去掉,不然后续会出错
在这里插入图片描述
上图点击下一步,出现以下界面cuda安装地址,这三个地址记一下,后面配置环境变量需要
在这里插入图片描述
然后下一步,等一会就安装完成了
在这里插入图片描述
之后验证cuda是否安装成功,首先查看安装目录下是都有nvcc可执行文件,及cudti64.dll,两者位置地址

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\bin\nvcc.exe
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\extras\CUPTI\lib64

如图:
在这里插入图片描述
在这里插入图片描述
然后在命令行中验证
在这里插入图片描述
2.cudnn下载
地址这里,找到相应版本,开始下载,下载完成后是压缩包如图,cudnn并非软件,而是一个工具库需要拷贝至cuda中
在这里插入图片描述
然后将压缩包解压,解压后得到cuda文件夹,将文件夹修改为cudnn然后考至cuda安装目录下,目录的话如果是默认c盘安装路径则在

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2

在这里插入图片描述
3. cuda环境变量的配置
打开修改环境变量界面,修改环境变量,添加path,将以下两地址添加并上移至cuda相关环境后面

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\extras\CUPTI\lib64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\cudnn\bin

在这里插入图片描述
4. 安装tensorflow-gpu
一开始按照这个链接安装这里安装的,没成功,可能我是哪里操作不对哈,然后我就没创建虚拟环境,直接在base环境中pip install --upgrade tensorflow-gpu安装成功了,下面是验证tensorflow-gpu版本安装成功的截图
在这里插入图片描述
到这里就安装完成tensorflow-gpu版本了,如有更便捷更好的安装方式,欢迎评论讨论哦,下一篇介绍pycharm的安装及配置~

  • 8
    点赞
  • 38
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值