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

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

相关推荐
Olafur_zbj4 分钟前
【AI】CUDA的新编程模型:tile编程模型 的好处
人工智能
Python大数据分析@30 分钟前
使用大模型MCP采集数据,爬虫已经无门槛
python·网络爬虫
Linguwen36 分钟前
外贸GEO01|GEO是什么?生成式引擎优化,AI时代的新流量密码
人工智能
物质波波波42 分钟前
WS-RPE:面向边缘物理AI实时特征值计算的硬件工作窃取调度器与冗余PE激活架构
人工智能·fpga开发·架构·系统架构·硬件架构
quantdash_cc43 分钟前
告别自建 Requests/BS4 网页爬虫:基于 QuantDash 搭建零维保的高性能量化行情流水线
开发语言·爬虫·python·pandas·量化·quantdash
起个名字好难啊这也被占用了1 小时前
LangGraphjs可中断可恢复的AI工作流
人工智能·ai编程
@Mr_LiuYang1 小时前
状态栏动态上下文信息追加到Agent --《深入理解 AI Agent :设计原理与工程实践》实验2-8
人工智能·大模型·动态上下文·状态栏信息追加
65岁退休Coder1 小时前
LangChain v1.3.4 笔记 - 07 补充:链式调用 LCEL
后端·python·langchain
卷无止境1 小时前
FastAPI 部署在 Nginx 后面到底该怎么配
后端·python