tensorflow / tensorflow-gpu cuda cudNN tensorRT 安装,启用显卡加速

news2024/11/24 12:10:20

tensorflow / tensorflow-gpu cuda cudNN tensorRT 安装,启用显卡加速

说明

Tensorflow-GPU 已被移除。请安装 tensorflow 。 tensorflow 通过 Nvidia CUDA 支持 GPU 加速操作。

自 2019 年 9月发布 的 TensorFlow2.1 以来,tensorFlow 和 tensorflow-GPU 一直是同一个包 。尽管校验和因元数据而异, 但它们以相同的方式构建,并且都通过Nvidia CUDA提供GPU支持。截至2022年12月,tensorflow-gpu已被移除,并已被这个新的空包所取代(https://pypi.org/project/tensorflow-gpu/),该包在安装时产生错误。
所有现有版本的 tensorflow-gpu 仍然可用,但 TensorFlow 团队已停止发布任何新的 tensorflow-gpu 包,并且 不会为现有的 TensorFlow-GPU 版本发布任何补丁。

所以大家直接安装 tensorflow ,安装好 NVIDIA 显卡驱动、CUDA 、CUDNN 、tensorRT 就可以启用GPU加速
没有特别的必要安装 Tensorflow-GPU这个包!!!
在这里插入图片描述

tensorflow-gpu 版本对应关系

tensorflow-gpu 版本对应关系 https://tensorflow.google.cn/install/source#gpu
在这里插入图片描述

安装方式

方式1 ,直接安装在系统上

这种方式只能安装一个版本
在这里插入图片描述

方式2,安装在虚拟环境中

可以安装多个虚拟环境对应LIB层不同版本,如图LIB层可以根据 tensorflow-gpu 版本对应关系选择,互不影响
在这里插入图片描述

1 安装nvidia显卡驱动

GPU计算能力 : https://developer.nvidia.com/cuda-gpus
GPU计算能力 > 3.5
驱动下载 :https://www.nvidia.com/Download/index.aspx

2 CUDA 安装

安装文档 https://docs.nvidia.com/cuda/
linux 安装文档 :https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
win 安装文档 : https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html

CUDA 各版本下载地址 : https://developer.nvidia.com/cuda-toolkit-archive

3 cudNN 安装

安装文档 : https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html

cudNN各版本下载地址 :https://developer.nvidia.com/rdp/cudnn-archive

4 tensorRT 安装

安装文档 https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html

各版本下载地址:https://developer.nvidia.com/tensorrt-download

5.1 tensorflow-gpu 安装

阿里源镜像: https://mirrors.aliyun.com/pypi/simple/tensorflow-gpu/
清华源镜像:https://pypi.tuna.tsinghua.edu.cn/simple/tensorflow-gpu/

5.2 或者 tensorflow 安装

阿里源镜像: https://mirrors.aliyun.com/pypi/simple/tensorflow/
清华源镜像:https://pypi.tuna.tsinghua.edu.cn/simple/tensorflow/

5.3 官方 conda 安装 tensorflow ,支持 cpu 和 gpu

linux 步骤: https://tensorflow.google.cn/install/pip#linux
win 步骤 : https://tensorflow.google.cn/install/pip#windows-native

conda 配置私有环境变量:

# CONDA_PREFIX 是当前conda 环境的路径 
mkdir -p $CONDA_PREFIX/etc/conda/activate.d

$CONDA_PREFIX/etc/conda/activate.d 路径下新建文件 env_vars.sh

添加以下内容:其中 安装路径 是自己的系统cuda cudnn tensorrt安装路径

echo " conda active export cuda cudnn tensorrt "
### cuda
export cuda_LIB_PATH=/安装路径/cudatoolkit-11.8.0/lib/
### cudnn
export CUDNN_LIB_PATH=/安装路径/cudnn-8.9.2.26-cuda11_0/lib/

# tensorRT
export tensorRT_PATH=/安装路径/TensorRT-8.6.1.6/lib

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$cuda_LIB_PATH:$CUDNN_LIB_PATH:$tensorRT_PATH

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/868501.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

NFT Insider#102:The Sandbox重新上线LAND桥接服务,YGG加入Base生态

引言:NFT Insider由NFT收藏组织WHALE Members(https://twitter.com/WHALEMembers)、BeepCrypto(https://twitter.com/beep_crypto)联合出品,浓缩每周NFT新闻,为大家带来关于NFT最全面、最新鲜、最有价值的讯息。每期周…

【JVM】类装载的执行过程

文章目录 类装载的执行过程1.加载2.验证3.准备4.解析5.初始化6.使用7.卸载 类装载的执行过程 类装载总共分为7个过程,分别是 加载,验证,准备、解析、初始化、使用、卸载 1.加载 将类的字节码文件加载到内存(元空间)中。这一步会…

实时时钟+闹钟

在江科大实时时钟的基础上添加闹钟的配置,参考http://t.csdn.cn/YDlYy。 实现功能 :每隔time秒蜂鸣器响一次、设置闹钟的年月日时分秒,到时间蜂鸣器响。 前三个函数没有变,添加 void RTC_AlarmInit(void) 闹钟的中断配置void…

分享Python技术下AutojsPro7云控代码

引言 有图有真相,那短视频就更是真相了。下面是三大语言的短视频。 Java源码版云控示例: Java源码版云控示例在线视频 Net源码版云控示例: Net源码版云控示例在线视频亚丁号-知识付费平台 支付后可见 扫码付费可见 Python源码版云控示例…

STM32CubeMX之freeRTOS消息通知(有点全能)

任务通知是任务自带的程序,不需要单独去创建 一: 二: 进入前不清除数据,退出清除数据参数 0x0000000000 0xffffffff的意思 三: 这里就是发送过去,然后把其存到了num中 不要有太多疑问,并不是发…

【Linux】TCP协议简介

TCP协议简介 TCP协议格式面向连接1.连接管理机制2.包序管理 可靠传输1.保证数据可靠到达对端2.保证数据的传输效率 面向字节流TCP粘包问题 TCP协议格式 16位源端口号和16位目的端口号:标识数据从哪个进程来,到哪个进程…

阿里云Windows服务器安装部署MySQL数据库流程

阿里云百科分享如何在Windows系统ECS实例上手动部署MySQL数据库。 目录 前提条件 操作步骤 前提条件 使用本教程进行操作前,请确保您已经注册了阿里云账号。如还未注册,请先完成账号注册。操作系统:Windows Server 2012准备一台ECS云服务…

JDK、JRE、JVM:揭秘Java的关键三者关系

文章目录 JDK:Java开发工具包JRE:Java运行环境JVM:Java虚拟机关系概述 案例示例:Hello World结语 在Java世界中,你可能经常听到JDK、JRE和JVM这几个概念,它们分别代表了Java开发工具包、Java运行环境和Java…

计算机丢失msvcr71.dll解决办法,总结三个常见的解决方法

修复msvcr71.dll文件的过程中,我对系统动态链接库文件的重要性有了更深入的了解。这个文件对于许多使用Visual C编译的软件来说是必不可少的,缺失或损坏可能导致软件无法正常运行。因此,当遇到类似问题时,及时解决并修复这个文件是…

页面文件太小,无法完成操作。

1、右键“我的电脑”,选择“属性”; 2、点击“高级系统设置”; 3、点击“高级”,再点击“设置”; 4、选择“高级”,选择“程序”,点击“更改”; 5、 不要勾选“自动管理所有驱动器…

Spring Boot+Mybatis实现增删改查接口开发+测试(超详细建议收藏)

前言 Java也是测试必知必会的内容,特别是现在类似spring boot 等Java框架更是成为主流。之前实现的图书增删改查是用Python实现的,没看过的请移步:Flaskmysql 实现增删改查接口开发测试(图文教程附源码),本…

教你如何使用AES对接口参数进行加密

教你如何使用AES对接口参数进行加密 前言 我们作为程序猿,在浏览网站的时候偶尔也会打开控制台看看请求的接口,我们会发现有些接口的传输是 “乱码” ,那么这个乱码究竟是什么呢?为什么要这么做? 其实这个所谓的 “…

无涯教程-Perl - qq函数

描述 可以使用此函数代替双引号。这实际上不是一个函数,更像是一个运算符,但是如果您在其他程序员的程序中看到它却不记得它是什么,那么可能会在这里看。实际上,您可以使用任何一组定界符,而不仅仅是括号。 语法 以下是此函数的简单语法- qq ( string )返回值 该函数返回双…

De Bruijin序列与魔术(三)——De Bruijin序列的拓展思考

早点关注我,精彩不错过! 在前面的文章中,我们已经介绍完经典DeBruijin序列的原理和魔术,相关内容请戳: De Bruijin序列与魔术(二)——魔术《De Bruijin序列》 De Bruijin序列与魔术(…

Chord diagram | 啧啧啧!~人人必会的Chord diagram你不来学一学吗!?

1写在前面 啊啊啊啊啊!!!!~终于值完夜班休息了。😭 最近是大搞医疗反腐的日子,㊗️各位执法人员成绩满满!~🤒 听说以后医务人员要年薪制了,完全搞不懂这些东西的初衷和理…

七夕好物分享,哪些礼物适合送男/女朋友?这几款好物最为合适!

七夕是个值得纪念的日子,牛郎织女鹊桥相会的故事百年流传,七夕是一个表达爱意的节日,送礼物是必不可少的,情侣们可以选择一份有意义的礼物,也可以选择对方需要的东西当做礼物来赠送,总的来说,送…

STM32F429IGT6使用CubeMX配置按键检测

1、硬件电路 2、设置RCC,选择高速外部时钟HSE,时钟设置为180MHz 3、配置GPIO引脚 4、生成工程配置 5、部分代码 /* USER CODE BEGIN 0 */ //按键检测函数 void KEY_Test(void) {if(SET HAL_GPIO_ReadPin(KEY1_GPIO_Port,KEY1_Pin)){while(SET HAL_GPIO_ReadPin(…

HCIP 链路聚合技术

1、链路聚合概述 为了保证网络的稳定性,仅仅是设备进行备份还不够,我们需要针对我们的链路进行备份,同时也增加了链路的利用率,提高带宽。避免一条链路出现故障,导致网络无法正常通信。这就可以使用链路聚合技术。 以…

PyTorch翻译官网教程-NLP FROM SCRATCH: GENERATING NAMES WITH A CHARACTER-LEVEL RNN

官网链接 NLP From Scratch: Generating Names with a Character-Level RNN — PyTorch Tutorials 2.0.1cu117 documentation 使用字符级RNN生成名字 这是我们关于“NLP From Scratch”的三篇教程中的第二篇。在第一个教程中</intermediate/char_rnn_classification_tutor…

走进知识图谱(二)【世界知识图谱篇】知识表示的经典模型与平移模型及基于复杂关系建模的知识表示学习

上篇文章提到&#xff0c;该系列文章将主要围绕世界知识图谱和语言知识图谱这两大类知识图谱进行展开&#xff0c;并且提到知识图谱的主要研究包括了知识表示学习、知识自动获取和知识的推理与应用三大部分。今天主要介绍世界知识图谱的知识表示学习&#xff0c;其中包括经典的…