Tensorflow使用GPU(cuda和cudnn和tensorflow下载)

提供一个比较简单的方法安装:不需要去那些网站下载cuda和cudnn,当然如果要升级系统cuda版本,还是需要重新安装的,升级可以看之前的文章

tensorflow和cuda和cudnn对应版本可以去官网查看

tensorflow和tensorflow-gpu的区别: TensorFlow 2.5+ 版本已合并 CPU/GPU 版本

2.5 及以上pip install tensorflow==x.x.x 单包包含 CPU/GPU 支持,自动适配环境

2.4 及以下pip install tensorflow-gpu==x.x.x 需单独装 GPU 版本,CPU 版是 tensorflow

本人电脑配置:win10 Navida cuda 是12.6

可以通过以下命令查看系统的cuda版本【注意我们在虚拟环境安装的cuda版本不能高于系统的】

复制代码
nvidia-smi

项目在pycharm运行,安装了Anaconda,后续每个项目在Anaconda创建一个虚拟运行环境

复制代码
# 1. 创建新环境(Python 3.8 是 TensorFlow 2.7 推荐的版本)
conda create -n tf27-gpu python=3.8 -y

# 2. 激活环境
conda activate tf27-gpu

# 3. 安装 CUDA 11.3 + cuDNN 8.2.1(如果还没装)
conda install cudatoolkit=11.3.1 cudnn=8.2.1 -c conda-forge -y

# 4. 安装 TensorFlow 2.7.0
pip install tensorflow==2.7.0

这里命令可以判断当前虚拟环境可以安装的cuda版本

复制代码
conda search cudatoolkit

可以通过以下命令判断cuda对应的cudnn版本

复制代码
conda search cudnn

顺序很重要,如果装错了,还是另外新建虚拟环境安装,多尝试

相关推荐
wechat_Neal2 小时前
告别敏捷2026:AI时代的软件开发新范式与生存指南
人工智能
新加坡内哥谈技术2 小时前
非常规 PostgreSQL 优化技巧在 PostgreSQL 中加速查询的创造性思路
人工智能
夜勤月2 小时前
连接 AI 的隐形纽带:深度解构 MCP 传输层——从 Stdio 到 SSE 的实战抉择与架构差异
人工智能·架构
徐小夕@趣谈前端2 小时前
【推荐】jitword协同文档新增AI公文助手,一键生成红头文件
vue.js·人工智能·开源·编辑器·github
深度学习lover2 小时前
<数据集>yolo毛毛虫识别<目标检测>
人工智能·python·yolo·目标检测·计算机视觉·毛毛虫识别
坐在地上想成仙2 小时前
从机床到键盘:用机械设计思维写出一个可部署网页
java·c++·python
Allen_LVyingbo2 小时前
用Python实现辅助病案首页主诊断编码:从数据清洗到模型上线(上)
开发语言·python·github·知识图谱·健康医疗
fantasy_arch2 小时前
SVT-AV1中的global-MV/仿射运动实现分析
人工智能·计算机视觉·av1
搬砖者(视觉算法工程师)2 小时前
【无标题】
人工智能·计算机视觉·机器人