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__)
相关推荐
桌面运维家1 分钟前
Windows防火墙高级配置:网络安全深度优化
windows·安全·web安全
吴佳浩 Alben2 分钟前
Vibe Coding 时代:Vue 消失了还是 React 太强?
前端·vue.js·人工智能·react.js·语言模型·自然语言处理
llm大模型算法工程师weng2 分钟前
Palantir 商业化关键时间点深度解析:从政府基本盘到 AI 爆发的战略跃迁
人工智能
飞哥数智坊3 分钟前
OpenClaw 中国行济南站圆满结束
人工智能
飞哥数智坊3 分钟前
openclaw 最近版本的崩溃与抢救
人工智能
起个名字总是说已存在3 分钟前
github开源AI Vibe Coding训练你的AI编程工具
人工智能·开源·github
饼干哥哥3 分钟前
OpenClaw真变态!我跑通了跨境电商的10个落地场景
人工智能
Mintopia4 分钟前
为什么同样写代码,有的人越写越轻松,有的人越写越乱
人工智能
hhzz6 分钟前
Openclaw案例之构建《全自动化、高适配、可定制”的AI绘画生产体系》
人工智能·ai作画·自动化·openclaw
飞哥数智坊6 分钟前
没有内测邀请码?我来帮你实测下 SOLO 网页端
人工智能·trae