Tensorflow-gpu windows安装

最新推荐文章于 2025-06-27 15:53:32 发布

原创最新推荐文章于 2025-06-27 15:53:32 发布 · 278 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#cuda #tensorflow #深度学习 #gpu

windows装机专栏收录该内容

1 篇文章

订阅专栏

该博客详细介绍了在Windows系统中安装Tensorflow-GPU2.2.0的步骤，包括先安装NVIDIA显卡驱动，然后下载并安装CUDA10.1和cudnn7.6。提供了CUDA和cudnn的百度云镜像链接，并指导将cudnn文件放入CUDA安装目录的相应文件夹。最后通过命令行验证安装成功，并使用pip3安装tensorflow-gpu2.2.0。

部署运行你感兴趣的模型镜像

Tensorflow-gpu windows安装

首先需要安装一个nvida的显卡驱动，这个用驱动精灵什么的都可以装
然后安装cuda，本文安装的是cuda10.1和cudnn7.6还有tensorflow-gpu2.2.0
先附上一个tensorflow和cuda和cudnn的版本对应图：
在这里插入图片描述
因此先去nvida官网下载这个cuda10.1，然后如果官网太慢，这里给个百度云的，cuda10.1的
链接：https://pan.baidu.com/s/1_1lvArUp8TS2vNzKmsG-Xg
提取码：vxw4
cudnn下载也是官网，需要注册一个账号才能下，这里也附上百度云吧
链接：https://pan.baidu.com/s/1pF9aZFkxMS3Jl4i4QVH3Ww
提取码：vy3t

随后，把cudnn里面的三个文件夹的文件放入c盘的cuda中对应的三个文件夹：bin、lib、include
随后安装完成，尝试命令行输入：nvcc -V
然后安装tensorflow-gpu就很简单了
一句话：pip3 install tensorflow-gpu==2.2.0

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理