在 anaconda 中安装 tensorflow models (gpu)

news2025/1/16 2:45:00

环境:Windows; Intel CPU + Nvidia GPU

1. 创建环境

  • 不推荐单次安装过多的库,可能导致安装失败(如超出终端缓存等)
  • 注意添加库的顺序 tensorflow-gpu 需要在 cudatoolkit 之前
  • 否则下载的 tensorflow-gpu 不支持 gpu 「实测」

TODO: 本文使用 mamba 平替 conda,旨在加速环境的构建速度
mamba 采用 C++ 重建了 conda 库,运行效率高于 conda,关于使用 mamba 提速 conda 相关内容可以参考:

mamba create -n tf-models-gpu-308 python=3.8 tensorflow-gpu cudnn cudatoolkit -c anaconda -c conda-forge

2. 添加提供 GPU 加速的库

  • keras 已包含在 tensorflow 中,故不再安装 keras-gpu
  • 手动添加 tensorflow-gpu,重复的库不会重复安装,同时可以避免被覆盖
mamba install tensorflow-gpu scikit-learn-intelex numba mkl-devel modin-all ipykernel -c anaconda -c conda-forge
  • 安装时,须留意 tensoflow-gpu 是否被覆盖成非 gpu 版本,如下图:(可以采用将 tensorflow-gpu 库置于首位的方式,避免覆盖)
    tensoflow

3. 添加常用的库

# 添加一些常用的库
mamba install scipy matplotlib scikit-learn pandas seaborn notebook -c anaconda -c conda-forge

虽然部分包之间相互依赖,但无需担心,conda 会自动处理的

4. 添加 tensorflow-model

tensorflow/models: https://github.com/tensorflow/models
TODO: 关于在 conda 中使用 pip 可以参考:

# 最后使用 pip 添加 tensorflow-model
# tensorflow-model 未提供 conda 安装方式,故使用 pip
pip3 install tf-models-official

5. 参考

  • Anaconda | Open Sourcing Anaconda Accelerate: https://www.anaconda.com/blog/open-sourcing-anaconda-accelerate
  • Anaconda | Getting Started with GPU Computing in Anaconda: https://www.anaconda.com/blog/getting-started-with-gpu-computing-in-anaconda
  • 关于 scikit-learn-intelex 提速: Anaconda | Scikit-learn Speed-up with Intel and Anaconda: https://www.anaconda.com/blog/scikit-learn-speed-up-with-intel-and-anaconda
  • Intel® Extension for Scikit-learn*— Intel® Extension for Scikit-learn* 2021.5 documentation: https://intel.github.io/scikit-learn-intelex/
  • 关于 Modin 提速 pandas: Anaconda | Scale your pandas workflow with Modin – no rewrite required: https://www.anaconda.com/blog/scale-your-pandas-workflow-with-modin
  • Installation — Modin documentation: https://modin.readthedocs.io/en/latest/getting_started/installation.html
  • Intel® Distribution of Modin: https://www.intel.com/content/www/us/en/developer/tools/oneapi/distribution-of-modin.html
  • Anaconda | Getting Started with GPU Computing in Anaconda: https://www.anaconda.com/blog/getting-started-with-gpu-computing-in-anaconda

  • 本文部分操作存在重复,但,是为了尽可能的确保 tensorflow gpu 版本不被覆盖
  • 署名:cc01cc: https://github.com/cc01cc
  • 欢迎大家转载分享,本作品采用署名-非商业性使用-禁止演绎 4.0 国际进行许可,转载请标明源地址,切莫修改或破坏原文结构,谢谢

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

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

相关文章

设备注册挂载流程(包含上电、使能、i2c通讯介绍)

目录 简介 上电时序 电压不同 时序不同 使能与复位 CLK时钟 I2C通讯 主从关系 识别设备 通讯格式 简介 任何相对于主板芯片的外挂设备都需要一定的注册挂载流程 (外挂设备:比如摄像头、nfc芯片、显示屏等等) 设备的挂载则需要满足…

JAVAEE-多线程(4)

目录 定时器 实现自己的Timer 线程池 常见的锁策略: 乐观锁和悲观锁 读写锁和普通互斥锁 重量级锁和轻量锁 自旋锁和挂起等待锁 公平锁和非公平锁 可重入锁和不可重入锁 synchronized CAS CAS和ABA问题 锁粗化 JUC 原子类 Semaphore CountDownLatc…

CAN总线控制器MCP2515 替代芯片 DP2515 DP2515-I/ST

汽车K总线与CAN的区别是什么 1、功能不同   K线一般用于检测系统,属单线模式,与诊断仪器连接并相互传递数据。CAN线主要用于控制单元与控制单元之间传递数据、属双线模式,分高位线和地位线。   2、通讯速度不同   K线通讯速率较低&…

101.对称二叉树 | 递归 + 迭代

对称二叉树 leetcode : https://leetcode.cn/problems/symmetric-tree/ 参考 对称二叉树 递归思路 首先在开始时, 一定要注意, 对称二叉树对比的并不是一个节点的左右子树, 而是两棵树, 这个很关键! 对比时是内侧和内侧对比, 外侧和外侧对比, 递归三步 : 确定递归的参数以…

1.1.2 了解JAVA语言

文章目录1 JAVA语言发展史2 面向对象的概念3 跨平台性4 JDK1 JAVA语言发展史 JAVA是由詹姆斯•高斯林(James Gosling)所创建的,其1977年获得了加拿大卡尔加里大学计算机科学学士学位,1983年 获得了美国卡内基梅隆大学计算机科学博…

4)Mybatis数据源以及事务实现

1. Mybatis数据源分为两种,一种直接连接数据库,一种使用连接池连接数据库,具体代码实现在包目录下 org.apache.ibatis.datasource 数据源接口: javax.sql.DataSource 池化数据源: org.apache.ibatis.datasource.…

OpenGL集锦(1)-安装与概述

目录概述fedora下安装编写OpenGL应用程序测试hello,world概述 OpenGL(英语:Open Graphics Library,译名:开放图形库或者“开放式图形库”)是用于…

Lichee_RV学习系列--CoreMark-Pro移植

Lichee_RV学习系列文章目录 Lichee_RV学习系列—认识Lichee Rv Dock、环境搭建和编译第一个程序 Lichee_RV学习系列—移植dhrystone 文章目录Lichee_RV学习系列文章目录一、CoreMark-Pro简介二、获取源码三、编译coremark-pro1、配置coremark-pro2、编译coremark-pro四、开发板…

各种树的总结

1.B树和B树 数据库的大量数据用什么存储?为什么是B树和B树?使用二叉树不行吗?先来说说他们的演变吧,首先如果用二叉树的话都为排好序的树查询起来是不是效率不高?所以此时我们提出了对树排序,就变成了二叉…

联想拯救者屏幕亮度无法调节,监视器和显卡驱动问题,经过多种测试

主要的问题位置 1,设备管理器中的监视器部分 2,设备管理器的显卡适配器部分 个人电脑出现这种情况的原因 自己拆一下机器加装固态,但这种感觉不应该导致问题。但导致这种问题的原因可能是装固态时候把电池拔了。 一些网上常说的方法 更新…

数字化转型对企业有什么意义?有哪些案例可以分享?

如何看待制造企业数字化转型?制造业企业数字化转型有哪些思路和案例? 一提到制造企业数字化转型,大多数人都认为,这是专属于大型制造企业的行为。其实不然,对于中小型制造企业,数字化转型也应该从易到难&a…

interview

1.PyTorch1.1 Conv2d1.2 dataset,dataloader1.3 训练pipeline1.4 梯度归零1.5 torch保存模型种类及区别2.目标检测2.1 yolo3,4,5,7区别2.2 yolo使用的loss(ciou,BCE等等)ciouBCElossL1,L2,CE,BCE2.3 图像增强2.4 IOU计算公式3.深度学习基础3.1 卷积公式4.TensorRT5.…

Niantic:未来AR重要场景,VPS众包3D地图到底是啥?

几个世纪以来,人们使用指南针、地图、星盘和象限仪来找路,而在过去二十年里,GPS成为了主流的定位系统,并且与手机结合后,让人们的出行越来越方便。而随着摄像头等技术发展,我们也开始看到视觉定位技术的崛起…

(almalinux,rockylinux,openeuler,openanolis,centos,ubuntu)云原生容器镜像漏洞trivy扫描对比

一、下载并安装trivy漏洞扫描工具 下载: https://github.com/aquasecurity/trivy/releases/download/v0.31.3/trivy_0.31.3_Linux-64bit.rpm 以下为centos平台的安装: [rootlocalhost ~]# rpm -ivh trivy_0.31.3_Linux-64bit.rpm Preparing... …

【算法刷题 DAY03】剑指offer树相关算法题总结2

JZ7 重建二叉树 描述 给定节点数为 n 的二叉树的前序遍历和中序遍历结果,请重建出该二叉树并返回它的头结点。 例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建出如下图所示。 提示: 1.vin.length pre.length 2.pre 和…

CSS入门二、美化页面元素

零、文章目录 文章地址 个人博客-CSDN地址:https://blog.csdn.net/liyou123456789个人博客-GiteePages:https://bluecusliyou.gitee.io/techlearn 代码仓库地址 Gitee:https://gitee.com/bluecusliyou/TechLearnGithub:https:…

【高速数字化仪应用案例系列】虹科数字化仪在光纤领域的应用

光纤应用 光纤越来越多地应用于各种领域。它们能够以光速长距离传输信息,并且损耗低,这使它们成为大容量远程数据通信的主要媒介。因此,光纤网络可以在电信系统中找到,它们用于传输和接收的目的。它们还用于提供各种数字服务&…

Docker命令-常用命令讲解

Docker常用命令 一:帮助命令二:镜像命令1. docker images 查看所有本地的主机上的镜像2. docker search 镜像名3. docker pull 下载镜像4. docker rmi三:容器命令1.docker run 新建容器并启动2.从容器返回到主机:3.docker ps 列出…

收银软件哪家强?2023年收银软件排行榜新鲜出炉!

每家实体店都少不了收银的程序,每个实体店老板都离不开收银的工具。随着信息技术的发展,收银的工具不再只有收银机,更高效、更方便的收银软件,已经成为了零售店老板们的新宠。收银机和收银软件有什么区别?收银机只能对…

1.11 LED灯点亮串口解析器

LED点灯实验 一.电路图: 三极管: NPN类型: PNP类型: NPN类型当基极为高电平时,集电极和发射极导通 PNP类型当基极为低电平时,集电极和发射极导通 由电路图可知LED电路图中三极管为NPN类型&am…