栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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 FilesNVIDIA GPU Computing ToolkitCUDAv11.2binnvcc.exe
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.2extrasCUPTIlib64

如图:


然后在命令行中验证

2.cudnn下载
地址这里,找到相应版本,开始下载,下载完成后是压缩包如图,cudnn并非软件,而是一个工具库需要拷贝至cuda中

然后将压缩包解压,解压后得到cuda文件夹,将文件夹修改为cudnn然后考至cuda安装目录下,目录的话如果是默认c盘安装路径则在

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.2


3. cuda环境变量的配置
打开修改环境变量界面,修改环境变量,添加path,将以下两地址添加并上移至cuda相关环境后面

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.2extrasCUPTIlib64
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.2cudnnbin


4. 安装tensorflow-gpu
一开始按照这个链接安装这里安装的,没成功,可能我是哪里操作不对哈,然后我就没创建虚拟环境,直接在base环境中pip install --upgrade tensorflow-gpu安装成功了,下面是验证tensorflow-gpu版本安装成功的截图

到这里就安装完成tensorflow-gpu版本了,如有更便捷更好的安装方式,欢迎评论讨论哦,下一篇介绍pycharm的安装及配置~

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/499170.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号