当前位置：网站首页 > 技术文章 > 正文

tensorflow GPU环境安装踩坑日记（安装tensorflow的gpu版最低配置）

网站名称：tensorflow GPU环境安装踩坑日记（安装tensorflow的gpu版最低配置）
网站分类：技术文章
收录时间：2025-07-08 14:39
网站地址：

进入网站

“tensorflow GPU环境安装踩坑日记（安装tensorflow的gpu版最低配置）” 网站介绍

前言：
最近做一个TensorFlow的开源项目，用CPU跑的话，要消耗太多的时间，于是有了这篇配置GPU环境的踩坑日志
分享一些注意的地方

去官网查看经过测试的构建配置

我使用的配置是

Win10+tensorflow2.1+CUDA10.1+cudnn7.6）

地址如下：
https://tensorflow.google.cn/install/source#linux

查看构建好的成熟的方案TensorFlow版本和CUDA和cudnn的版本之间的关系

1.安装tensorflow

使用conda创建一个虚拟环境

！！注意！！

pip install -i https://pypi.doubanio.com/simple tensorflow=2.1.0

2.安装cuda10.1

下载地址：

https://developer.nvidia.com/cuda-toolkit-archive

一路next安装完成之后

查看环境变量，默认cuda的环境变量会自动添加

配置path中的环境变量

在终端中测试：

3.cudnn7.6配置文件

文件下载：
https://developer.nvidia.com/rdp/cudnn-archive

下载这个文件需要登录NVIDIA官网，我登录了半天登不上去，后面把网页的语言设置成English登上去了

把压缩包解压出来

测试能不能使用GPU

import tensorflow as tf import os os.environ['TF_CPP_MIN_LOG_LEVEL']='2' a = tf.constant(1.) b = tf.constant(2.) print(a+b) print('GPU:', tf.test.is_gpu_available())

GPU环境成功会打印如下

cuda10.1下载

上一篇：1、从零开始了解和使用WPS的js宏(JSA)
下一篇：MinerU私有化部署指南:构建安全的文档智能解析引擎

tensorflow GPU环境安装踩坑日记（安装tensorflow的gpu版最低配置）

“tensorflow GPU环境安装踩坑日记（安装tensorflow的gpu版最低配置）” 网站介绍

1.安装tensorflow

2.安装cuda10.1

3.cudnn7.6配置文件

更多相关网站