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__)
相关推荐
2501_94583743几秒前
OpenClaw:让 AI 从 “聊天” 到 “干活” 的本地智能体革命
人工智能
实在智能RPA3 分钟前
Agent 如何处理流程中的异常情况?——2026企业级智能体稳定性架构深度拆解
开发语言·人工智能·ai·架构·php
AIoT科技物语4 分钟前
免费开源!50+算法,Java基于YOLO框架的视频AI识别算法平台,适配低空无人机巡检、摄像头安防场景
java·人工智能·算法·yolo·开源
翔云1234565 分钟前
一文读懂人工智能,机器学习,深度学习,神经网络,Transformer
人工智能·深度学习·机器学习
盛世宏博北京5 分钟前
多协议温湿度传感器技术解析及系统集成方案设计
大数据·人工智能·温湿度传感器
sakana5 分钟前
如何写一个自己的skill
前端·人工智能
FONE_Platform6 分钟前
FONE亮相第十二届CDIE数字化创新博览会
人工智能·ai·cdie
编程百晓生7 分钟前
《SAP FICO系统配置从入门到精通共40篇》021、CO-PC实战笔记:在制品与差异计算,那些年车间里的“账实不符”
服务器·人工智能·笔记·ar·从入门到精通·sap fico·sap fico系统
weixin_4634497813 分钟前
智媒时代,如何借力权威发声?——探秘官媒发布系统的变革力量
人工智能·自动化
慕涯AI18 分钟前
Agent 30 课程开发指南 - 第19课
人工智能·python