win10+1050Ti+tensorflow-gpu2.2.0+cudatoolkit10.1+cudnn7.6.5配置Tensorflow-gpu框架(无须本地安装cuda+cudnn)

news/2025/3/5 5:28:49/

避坑安装Tensorflow-gpu

  • 安装心理路程(吐槽)
  • 安装环境选择
  • 安装过程详解
  • 测试

安装心理路程(吐槽)

  历时两天时间,终于将Tensorflow-gpu安装成功, 期间观看不下于50篇的博客,也尝试多种的安装方法,其中电脑驱动就来回捣鼓数次,然后就是cuda和cudnn下载了多个版本,现在后悔它们都没用了,还有Python的版本问题也捣鼓了许久,呜呜呜(心疼手机流量)…最后我重装的系统,将win10的企业版换成了专业版,专业版一次成功,但应该和电脑系统没啥关系…个人猜测
  虽然失败了很多次,期间也想过放弃,使用CPU跑,但笔记的cpu跑深度学习,大规模数据和复杂网络结果出的真的慢,还是GPU香,废话不多说了,开干…
总结目前安装tensorflow-gpu主要是两种途径:
① 直接本地安装(cuda+cudnn,不推荐,如何第一次没成功卸载特别难受)
② 利用conda安装(cudatoolkit+cudnn,推荐,贼舒服)

安装环境选择

使用群体:使用win10的进行深度学习的同学,安装ubuntu(linux)的同学可以不用看了

  1. 整个环境的介绍:
    ① 安装Anaconda3:目的创建虚拟环境
    ② vscode:进行Python code,方便
    ③ 框架: Tensorflow

  2. 版本选择:
    ① NVIDIA显卡驱动:可以为最新
    ② Anaconda3: 直接最新版,Anaconda3支持python版本向下兼容
    ③ vscode:直接最新版
    ④ python环境:python 3.7(这个待会创建虚拟环境会设置)
    ⑤ cudatoolkit版本:10.1(重点)
    ⑥ cudnn版本:7.6.5(重点)
    ⑦ tensorflow-gpu:2.2.0

  3. 补充说明
    ① 官网说tensorflow2.1版本后tensorflow中包含GPU,但我没有试过,我安装仍然是tensorflow-gpu,感兴趣可以重创环境试下
    tensorflow-gpu和cudatoolkit和cudnn版本要对应,否则会出现意想不到的错误,下图是目前的最新版本对应:
    在这里插入图片描述

    查询Tensorflow版本对应的: tensorflow官方技术文档
    ③ 注意自己的电脑驱动支持的cuda版本,查看到支持版本一定大于等于将下载的版本:
    在这里插入图片描述
    上图中,表明我的电脑驱动版本为457.20,支持的最高版本的cuda为11.1(也就是我能安装的cuda不能高于11.1,如果想装更高的版本cuda,可以到官网下载驱动,更新自己的驱动)

安装过程详解

在这里插入图片描述
贴下我的配置依据

  1. 安装Anaconda3
  2. 更新下自己的conda的默认镜像源
  3. 创建虚拟环境:

conda create -n tensorflow2.2(名字,可修改) python=3.7
conda env list # 查存在的环境
conda activate tensorflow2.2 #进入虚拟环境

  1. 安装tensorflow-gpu2.2.0版本,使用pip安装,采用豆蔻镜像

pip install tensorflow-gpu==2.2.0 -i https://pypi.douban.com/simple/

  1. 安装cudatoolkit10.1+cudnn7.6.5

conda install cudatoolkit=10.1 cudnn=7.6.5

至此安装工作结束,推荐大家完全按照上述指令在cmd中进行操作,等成功后可尝试其他版本。

测试

  1. 查看是否安装成功,是够存在GPU
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

在这里插入图片描述
2. 测试GPU是否能运行

import tensorflow as tf
import timeit
import osos.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'  # 代码用于忽略级别 2 及以下的消息(级别 1 是提示,级别 2 是警告,级别 3 是错误)。with tf.device('/cpu:0'):cpu_a = tf.random.normal([10000, 1000])cpu_b = tf.random.normal([1000, 2000])print(cpu_a.device, cpu_b.device)
with tf.device('/gpu:0'):gpu_a = tf.random.normal([10000, 1000])gpu_b = tf.random.normal([1000, 2000])print(gpu_a.device, gpu_b.device)
def cpu_run():with tf.device('/cpu:0'):c = tf.matmul(cpu_a, cpu_b)return c
def gpu_run():with tf.device('/gpu:0'):c = tf.matmul(gpu_a, gpu_b)return c
# warm up
cpu_time = timeit.timeit(cpu_run, number=10)
gpu_time = timeit.timeit(gpu_run, number=10)
print('warmup:', cpu_time, gpu_time)cpu_time = timeit.timeit(cpu_run, number=10)
gpu_time = timeit.timeit(gpu_run, number=10)
print('run time:', cpu_time, gpu_time)print('GPU', tf.test.is_gpu_available())

运行结果:
warmup: 1.0394134999999993 0.2286729000000003
run time: 1.0295364999999999 0.0005075000000003271
GPU True
通过运行时间可以看出,说明tensorflow-gpu安装成功

祝大家都能成功,享受tensorflow-gpu的快乐!!!!!


http://www.ppmy.cn/news/244105.html

相关文章

【CUDA for GTX一次点亮】winows10 + GTX1050Ti + tensorflow1.8 + CUDA9.0 + cudnn7.1

按如下步骤,一次安装成功CUDA9.0 for GTX。 搭配GPU,计算速度果然飞速提升。 O. 本人计算机及软件配置: i7-8750H 、GTX1050Ti 、Anaconda3 -64bit (安装时选择了将路径添加到环境变量,也选择了下载并安装VS&#xff…

TFLite文件解析及格式转换

随着深度学习越来越流行,工业生产不光在PC端应用场景丰富,在移动端也越来越凸显出深度学习的重要性及应用价值。由于嵌入式平台受存储、指令集限制,需要提供更小的网络模型,并且某些DSP平台不支持float指令。tensorflow提供TOCO转…

iOS 高级工程师面试必备

请简要介绍一下 MVC、MVVM 和 VIPER 架构模式。它们的优缺点分别是什么? MVC(Model-View-Controller): MVC 是一种经典的软件架构模式,主要分为三个部分:Model(模型)、View&#xf…

CS1237 数据实测

目录 硬件 测试数据 结论 性能 转换时间 硬件 外部电阻电位器 模拟 传感器信号 测试数据 一下记录数据为ADC原始数据未经过软件滤波 跳动幅度图片DEV_FREQUENCY_10DEV_PGA_1751DEV_FREQUENCY_40DEV_PGA_11562DEV_FREQUENCY_640DEV_PGA_18243DEV_FREQUENCY_640DEV_PGA_644…

Linux内核sync流程

进程写文件时, 文件并没有真正写到存储设备, 而是写到了page cache中。 文件系统会定期把脏页写到存储设备, 进程也可以调用sync 这样的调用把脏页写回存储设设备。 数据结构 backing_dev_info 要理解这个结构体, 得从它需要解…

牛客网Linux错题六

1.有一个文件ip.txt,每行一条ip 记录,共若干行,已排好序,下面哪个命令可以实现“统计出现次数最多的前3个ip及其次数”?(B) A. uniq -c ip.txt B. uniq -c ip.txt | sort -nr | head -n 3 C.…

SpringBoot个人博客系统(含源码+数据库)

一、作品设计理念 个人博客系统是一个让个人可以通过互联网自由表达、交流和分享的平台,是个人展示自己思想、感受和经验的品牌。设计理念对于任何一个个人博客系统来说都非常重要,它直接影响到用户的使用体验和网站的整体感觉。 好的设计理念应该着眼于…

预见未来:超强元AI诞生,抓住这个机会,利用AI变现也变得更加容易

目录 一、引言 二、介绍 三、技术展现 四、元AI架构图展现 五、元AI变现技巧—商业版说明 六、后期规划 一、引言 如何利用AI变现已经成为了当今各个行业亟需解决的问题。随着人工智能技术的快速发展和普及,越来越多的企业开始将其应用于产品研发、销售流程优化、客…