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 天前
GigE Vision 多相机同步终极检查清单(可直接用于项目部署)
开发语言·人工智能·数码相机·机器学习·计算机视觉·视觉检测·工业相机
wenjingdadi1 天前
自学小模型day2——YOLO模型的输出指标
人工智能·yolo·机器学习
AugustRed1 天前
AI流式输出方案SSE vs WebSocket对比
人工智能·websocket·网络协议
砍材农夫1 天前
spring-ai 第五模型介绍
java·人工智能·spring
DeepModel1 天前
机器学习降维与信号分离:独立成分分析 ICA
人工智能·机器学习
派大星~课堂1 天前
【力扣-94.二叉树的中序遍历】Python笔记
笔记·python·leetcode
SQVIoMPLe1 天前
python-langchain框架(3-7-提取pdf中的图片 )
python·langchain·pdf
Ulyanov1 天前
音视频分离与音频处理核心技术深度解析 从MP4到高品质音乐文件的完整技术实现
python·音视频
阿勉要睡觉1 天前
深度学习的模型量化
人工智能
KuAI_KST1 天前
医美抖音AI客服推荐,低成本撬动本地医美获客增长
人工智能