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 分钟前
2026全国AI大模型备案代办机构口碑排行及选型参考
人工智能
码农学院3 分钟前
本地服务业GEO架构方案:用 ASP.NET Core 中间件按门店动态注入 LocalBusiness Schema,营业时间改一次全站生效
人工智能·geo优化·ai优化aio
user_admin_god4 分钟前
第 05 篇:结构化输出 —— 让模型稳定返回 JSON
java·人工智能·spring boot·语言模型
moonsims4 分钟前
AiBrainBox-UGV 的目标跟踪从“传统视觉跟踪”升级为“语义目标跟踪”
前端·人工智能·量子计算
AI你一生一世7 分钟前
当手机镜头遇上实时流:移动端4K直播的技术突围与选型指南
人工智能·音视频编解码·技术选型·4k视频·移动端直播·实时流媒体
知几蜗牛7 分钟前
100万Token不等于模型全记住:从KV Cache看懂长上下文成本
人工智能
yangmu320313 分钟前
RTX 40系显卡性能终极解锁:OptiScaler开启DLSS 5与6倍帧生成硬核指南
人工智能·游戏程序
高升说16 分钟前
移动机器人避障相机安装与视场覆盖:一套可核算的工程方法
人工智能
deepseek2319 分钟前
RSIAgent 拆解:不更新模型参数,Agent 靠环境自探索把开源模型推过 GPT-6 Astra,Scaling Experience 的递归自提升
人工智能·ai agent·开源模型