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__)
相关推荐
weixin_446260857 小时前
从被动镜像到主动智能体:面向网络物理人工智能的整体论数字孪生(HDT-Net)
网络·人工智能
秋枫要学习7 小时前
你的鼠标,正在被 AI 抢走
人工智能·计算机外设
满怀冰雪8 小时前
19-图像数据处理:PaddleVision Transform 实战
人工智能·深度学习·计算机视觉·paddle
IT_陈寒8 小时前
Vue的响应式让我原地破防,原来问题出在这
前端·人工智能·后端
带娃的IT创业者8 小时前
Kimi-K3 开源背后:2.8 万亿参数的“暴力美学”与智能体的新拐点
人工智能·开源·大模型·智能体·开源模型·kimi-k3·moonshot ai
m0_380743879 小时前
实战看出 LLM API 的隐性消耗
人工智能
安逸Ai9 小时前
Claude Code、Cursor、Copilot 这类工具通常如何理解项目上下文?
人工智能
烟雨江南7859 小时前
2026汽车制造与新能源整车柔性产线Agentic-Workflow白皮书:跨APS_MES_ERP多智能体动态排产与装配容错实战
大数据·网络·人工智能·自动化·ai客服·企业agent
星栈10 小时前
AI 时代,人和 AI 的关系,我的一点思考
人工智能
鲜于言悠90510 小时前
Claude Code重大更新:多会话可互相通信,告别手动复制上下文
人工智能