windows环境下配置tensorflow_gpu版本——无需更改本地的cuda

大家可以在tensorflow学习中,可能会遇到使用tensorflow_gpu版本的安装,但是一般涉及到gpu的安装,就需要配置cuda,这个过程很麻烦且浪费时间,下面给出一个简单的方法配置环境。

假设已经创建好虚拟环境,这里我们选取tensorflow-gpu==2.1.0为例进行举例,注意python版本在3.6-3.9之间

可以看出需要cudnn=7.6,cudatoolkit=10.1

安装cudnn

python 复制代码
conda install cudnn=7.6.5

安装cudatoolkit

python 复制代码
conda install cudatoolkit=10.1

安装tf=2.1.0

python 复制代码
pip install tensorflow-gpu==2.1.0

如果自己pip或者conda下载慢的话,可以直接下载对应的whl包,pip安装就行。

网址放在这 https://www.tensorflow.org/install/pip?hl=zh-cn ,自行下载即可

验证是否安装成果

在终端输入以下命令

python 复制代码
import tensorflow  as tf
print(tf.__version__)
相关推荐
梦醒江南14 小时前
从调参到调水:OpenClaw深度技术指南——如何科学“饲养”你的AI小龙虾
人工智能
穿条秋裤到处跑814 小时前
java2AI系列:SpringAI 通过 Function Calling 接入外部系统
java·人工智能
_遥远的救世主_14 小时前
每个软件都需要一个 CLI + Skill:未来软件架构的终局猜想
人工智能
qq_3813385014 小时前
2026 年 AI 编程助手全面对比评测:Cursor vs Copilot vs Claude Code vs GitHub Copilot Free
人工智能·github·copilot
W_Pan14 小时前
我给 miniONNXRuntime 做了 mac + CUDA 的 EP,这里是实现思路
人工智能
东离与糖宝14 小时前
医疗辅助 Agent:病历解读、文献检索、指南对照
人工智能
天一生水water14 小时前
机器学习中的小提琴图有什么作用
人工智能·机器学习
古城小栈14 小时前
Rust在当下AI领域的用武之地:从底层加速到上层应用全解析
开发语言·人工智能·rust
一次旅行14 小时前
2026最新AI编程模式总结
人工智能
东离与糖宝14 小时前
泛化能力基础:AI 适应新数据的关键
人工智能