当前位置: 首页 > news >正文

win11/win10+tensorflow2.9.0+cuda11.2.1+cudnn8.1.1

win11/win10+tensorflow2.9.0+cuda11.2.1+cudnn8.1.1

  • 1.查看显卡支持的最高cuda版本
  • 2.cuda安装
  • 3.cudnn
  • 4.Tensorflow-GPU
    • 4.1创建一个新的conda环境
    • 4.2 测试

1.查看显卡支持的最高cuda版本

我是win11系统

查看显卡在这里插入图片描述
在这里插入图片描述
所以只要下载的cuda低于查看的版本,显卡驱动就支持,向下兼容。
但是要注意

  • CUDA 即英伟达的显卡并行计算框架,nvidia-smi 可以查看
    tensorflow-gpu的运行需要它的底层支持,它是一个计算框架,抽象层次比驱动高,每个版本的CUDA都是基于一定版本的驱动建立的,所以它对驱动的最低版本是有要求的
  • cuDNN 基于CUDA架构的神经网络库
    是专门用于神经网络计算的加速包,可以看作是CUDA的一部分(并行计算框架下的一个包)
  • tensorflow-gpu
    tensorflow-gpu的运行需要用到CUDA框架, 特别是cuDNN库,就像CUDA是基于特定版本驱动构建一样,tensorflow-gpu也是调用特定版本的CUDA,二者存在对应关系
显卡驱动决定了CUDA的最高版本

CUDA决定了cuDNN的版本

CUDA决定了tensorflow-gpu的版本

tensorflow-gpu决定了python的版本
进cmd
输入nvidia-smi

在这里插入图片描述
可以看到TensorFlow-gpu对应的cuda最高版本为11.2,但是点进去发现最高只支持win10系统,没有win11的选项。最后实测win11也是向下兼容的,直接开整。
在这里插入图片描述

2.cuda安装

cuda官网链接:https://developer.nvidia.com/cuda-toolkit-archive/
我选择的版本是11.2.1
在这里插入图片描述
可以看到没有win11选项,直接选中win10就行。
在这里插入图片描述
下载完毕后直接一路next安装。
安装好后cmd命令行中测试一下

cmd
nvcc --version

在这里插入图片描述

3.cudnn

cudnn官网链接: https://developer.nvidia.com/rdp/cudnn-archive#a-collapse51b
我选择的是8.1.1版本
在这里插入图片描述
在这里插入图片描述
下载好了解压后,将cudnn里面的bin、include、lib文件夹所有内容复制到对应的cuda文件夹C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1下面。
在这里插入图片描述
在这里插入图片描述

4.Tensorflow-GPU

4.1创建一个新的conda环境

conda create --name tf22 python=3.8
activate tf22
pip install tensorflow-gpu==2.9.0 -i https://pypi.mirrors.ustc.edu.cn/simple

python版本我这里选择3.8,tensorflow-gpu一定要指定好版本。
在这里插入图片描述
pip安装比较慢,可以使用镜像

中科大镜像:https://pypi.mirrors.ustc.edu.cn/simple

豆瓣镜像:http://pypi.douban.com/simple

阿里镜像:https://mirrors.aliyun.com/pypi/simple

百度镜像:https://mirror.baidu.com/pypi/simple

激活刚刚创建的环境
在这里插入图片描述
在创建的环境中直接使用pip安装,建议使用镜像安装,速度很快。
在这里插入图片描述

4.2 测试

使用PyCharm简单测试,注意如果是中途修改过环境依赖中的东西,例如重新安装过TensorFlow,要重启PyCharm,重新导入依赖才能生效,否则会出现已经成功安装但是却输出False情况。

import tensorflow as tf
print(tf.version)
print(tf.config.list_physical_devices('GPU'))
print(tf.test.is_gpu_available())

在这里插入图片描述

查看结果,可以看到输出了显卡的有关信息,表示安装成功。如果最后没有出现True和显卡信息,则说明中间过程可能出现了问题,可以先试着重启一下
PyCharm再试试。在这里插入图片描述
最后也有人使用高版本cuda11.7成功了,也可以参考。
win11+cuda11.7+cudnn8.5+Tensorflow-GPU : https://www.cnblogs.com/LandWind/p/win11-cuda-cudnn-Tensorflow-GPU-env-start.html

相关文章:

  • leetcode日记(77)子集Ⅱ
  • vue3:初学 vue-router 路由配置
  • 备份docker的数据库文件信息
  • PMP项目管理—资源管理篇—1.规划资源管理
  • 【Python爬虫】爬取公共交通路网数据
  • 大语言模型 智能助手——既能生成自然语言回复,又能在必要时调用外部工具获取实时数据
  • 【零基础到精通Java合集】第二十二集:CMS收集器详解(低延迟的里程碑)
  • 将 MySQL 数据高效导入 Redis
  • JAVA安全—Shiro反序列化CB1链source入口sink执行gadget链
  • Python解决“找出整形数组中占比超过一半的数”问题
  • AI介入软件测试过程可行的环节和帮助机制
  • 科普:“图相似性”与“文本相似性”
  • 【音视频】ffmpeg命令分类查询
  • LabVIEW设备长期稳定运行程序设计要点
  • MySQL安装教程
  • 【数据结构】二叉树总结篇
  • 二百八十五、华为云PostgreSQL——建分区表并设置主键
  • *动态规划(4)
  • 精讲坐标轴系统(Axis)
  • 实现浏览器交互Ai Web Ui-本地化部署的deepseek + Ollama + Page Assist
  • 网站设置手机版/流量平台有哪些
  • 网站和域名的关系/关键词挖掘站长
  • 英文网站建设模板下载/湖南seo优化哪家好
  • 网站域名证书/长春关键词优化平台
  • ic电子网站建设/网络推广有哪些常见的推广方法
  • 网站制作公司怎么收费/360关键词指数查询