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

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

相关推荐
一点一木1 天前
深度体验TRAE SOLO移动端7天:作为独立开发者,我把工作流揣进了兜里
前端·人工智能·trae
Lee川1 天前
mini-cursor 揭秘:从 Tool 定义到 Agent 循环的完整实现
前端·人工智能·后端
weelinking1 天前
【产品】00_产品经理用Claude实现产品系列介绍
数据库·人工智能·sql·数据挖掘·github·产品经理
Agent产品评测局1 天前
制造业模具管理AI系统,主流产品能力对比详解:2026年智能制造选型深度洞察
人工智能·ai·chatgpt·制造
2301_803934611 天前
Go语言如何做网络爬虫_Go语言爬虫开发教程【指南】
jvm·数据库·python
WL_Aurora1 天前
Python爬虫实战(六):新发地蔬菜价格数据采集.
爬虫·python
盲敲代码的阿豪1 天前
Python 入门基础教程(爬虫前置版)
开发语言·爬虫·python
研华科技Advantech1 天前
如何用一套实训设备,打通工业AI预测性维护技术全流程?
人工智能
Lab_AI1 天前
AI for Science: MaXFlow AI Agent+ 报告体验双升级,让AI智能体更高效易用!
人工智能·ai for science·ai agent·ai智能体
李坤1 天前
让 Codex 和 Claude 互相 Review:告别手动复制
人工智能·openai·claude