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

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

相关推荐
电子元件小说家18 分钟前
2026年四季度电感市场观察:AI算力与汽车电子驱动下的选型与验证建议智能硬件
人工智能·汽车
网易CodeWave-小码哥21 分钟前
支持多模块 Java 微服务导入|CodeWave SDD V4.5.10 上线
人工智能
AI你一生一世33 分钟前
不止于时序存储:DolphinDB 如何成为 AI 时代的工业智能底座
java·人工智能·struts·时序数据库·工业智能·dolphindb·ai基础设施
belldeep36 分钟前
python:genesis-world, Genesis-Embodied-AI
人工智能·python·ai·genesis
怕浪猫1 小时前
小程序成长计划上新:10 亿 Token 现在能调 DeepSeek-v4.1-flash了
人工智能·aigc·ai编程
ss2731 小时前
为什么PostgreSQL适合AI场景?
数据库·人工智能·postgresql
benchmark_cc1 小时前
A 股、港股、美股如何建立统一的标的主数据格式?
开发语言·python·数据分析·pandas·量化交易·股票数据·quantdash
七牛云行业应用2 小时前
DeepSeek Super-agent 与 Grok Bot:开源 Agent 架构和聊天助手有何区别
人工智能·ai·架构·ai编程
维克兜率天2 小时前
【维克】趋势型CTA:期货市场的“追涨杀跌“
人工智能·python
u1301302 小时前
GitHub 热榜项目:日榜(2026-11-03)
人工智能·github