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__)
相关推荐
redreamSo4 分钟前
团队 AI 资产总烂在本地?先分清哪些该装、哪些只能连
人工智能·架构·开源
MacroZheng4 分钟前
短短几天,暴涨2.8万Star!又一款编程神器开源!
java·人工智能·后端
东风破_7 分钟前
Harness Engineering:让大模型从“会回答”走向“能干活”
人工智能
武子康9 分钟前
调查研究-208 OpenAI GPT-5.6 Sol / Terra / Luna 解读:AI 模型竞争正在从“更聪明“转向“能长期干活“
人工智能·openai·claude
达达尼昂11 分钟前
Claude : 如何设计可控的agent-loops
前端·人工智能·后端
武子康22 分钟前
调查研究-207 Claude Sonnet 5 发布:Agent 能力下放后,模型路由要从“强弱分层“改成“执行分层“
人工智能·agent·claude
IT_陈寒41 分钟前
Redis持久化丢失数据的坑,这次终于被我填平了
前端·人工智能·后端
树獭非懒2 小时前
从零构建ReAct智能体:让AI学会边想边做
人工智能·llm·agent
冬奇Lab13 小时前
Workflow 系列(04):Multi-Agent 协调——编排器边界、并发控制与上下文隔离
人工智能·工作流引擎
冬奇Lab13 小时前
每日一个开源项目(第147篇):HyperGraphRAG - 用超图表示 N 元关系,RAG 的第三代范式
人工智能·开源·graphql