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__)
相关推荐
hpoenixf3 分钟前
Agent 运行身份全景:从一次聊天到 Model Turn,系统到底在继续什么?
人工智能
玩美数据12 分钟前
市场调研数据分析与数据集驱动的行业格局变革深度解析
大数据·人工智能
lucas_AI22 分钟前
AI 思考时非要「说人话」吗?哈工大团队让模型「用数据格式思考」,省 5 倍 token 还更准
人工智能
知行产研25 分钟前
紫创智械:工程机械具身智能的「非运输」破局者
人工智能
微订校园外卖系统1 小时前
校园代取快递系统选型:现成系统与定制开发的技术核对项
大数据·运维·人工智能·外卖系统·本地生活·微订
QXWZ_IA1 小时前
管道GIS测绘如何实现自动化成图:三种成图方式怎么选
运维·人工智能·安全·自动化
严肃小白兔1 小时前
第 04/24 期:深度学习中的损失函数及其数学解释
人工智能·深度学习
无责任此方_修行中1 小时前
「不止技术」AIE 2026 会后感悟
人工智能·后端·ai编程
BYSJMG2 小时前
计算机毕业设计选题推荐:【基于深度学习的车辆损坏严重程度评估方法研究】ResNet50 图像分类
人工智能·深度学习·目标检测·机器学习·计算机视觉·分类·课程设计
在所不辞兄2 小时前
为什么在电池供电场景下,不能采用固定电压频率策略,而需要动态调整以最大化 Race-to-Sleep 收益?
人工智能·深度学习·神经网络·机器学习