news 2026/9/3 0:19:25

深度学习 SOP: conda通过命令快速构建指定版本tensorflow gpu环境。

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
深度学习 SOP: conda通过命令快速构建指定版本tensorflow gpu环境。

准备工作

在这个SOP中,我不会指定任何一个版本,而是指导你如何根据你的需要的版本取去查询对应的依赖版本!这样能让这个SOP持续可用!

在开始安装之前,我们需要了解三个关键版本,这决定了环境是否能兼容运行。

  • Tensorflow版本
  • CUDA版本
  • cuDNN版本

不用担心复杂,官方已经整理好了依赖关系,你需要做的就只有确保你的电脑已经正确安装了GPU驱动,这可以通过命令nvidia-smi来进行查看,在命令结果的右上角,你可以看到你驱动所支持的最大CUDA版本。


步骤

1. 前往TensorFlow官网查看目标版本所需依赖

经过测试的构建配置
在上方这个链接里TensorFlow官方给出了不同TensorFlow版本所兼容的CUDA和cuDNN版本。

假如现在团队要求安装Tensorflow 2.3.0版本。我们就找到这一行:

版本Python 版本编译器构建工具cuDNNCUDA
tensorflow-2.3.03.5-3.8GCC 7.3.1Bazel 3.1.07.610.1

取出我们所需的三个关键版本

  • Python版本:3.8
  • CUDA版本:10.1
  • cuDNN:7.6

当然也可以根据你的需要选择版本,主要就是取到这三个关键版本,Python版本一般选支持范围最新的就可以。

2. 创建conda环境并激活

替换team-tf23为你想创建环境名称、替换3.8为Python版本。

conda create -n team-tf23python=3.8-y conda activate team-tf23

3. 安装环境独立的CUDA/cuDNN

替换10.17.6为你需要的CUDA/cuDNN版本。

condainstall-ycudatoolkit=10.1cudnn=7.6

*如果慢且长时间安装不成功,请参考附录的解决方案。

4. 安装 TensorFlow

替换2.3.0为你需要的TensorFlow版本

pipinstall"tensorflow==2.3.0"

这里建议直接装 tensorflow==2.3.0。原因是 从 TF 2.1 起,tensorflow 与 tensorflow-gpu 本质上是同一个包(后来官方也明确让大家装 tensorflow)。

*如果感觉慢,或者报错433。可以尝试临时使用国内源(这里是清华源)。

pipinstall"tensorflow==2.3.0"-i https://pypi.tuna.tsinghua.edu.cn/simple

附录:

如果在第三步中感觉慢,可以尝试临时使用国内源(这里是清华源)。

condainstall-y\cudatoolkit=10.1\cudnn=7.6\-c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge\--override-channels

或者长期更换conda源。清华源Conda换源教程
首先查看现有源。

conda config --show-sources

一般是只有一个default,通过这个命令添加自定义源(这里是清华源)。

conda config --set custom_channels.auto https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/

再次查看现有源,可以看到已经添加了清华源,此时可以再次尝试安装。

conda config --show-sources

如果出现错误,可以用这个命令移除。

conda config --remove channels 源名称

总结

本文的主要场景是针对团队合作、项目复现、导师要求等情况下需要安装【指定版本】的SOP。当然安装最新版本也可以按照这个流程自行调整。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/8/27 8:51:20

DeepSeek-OCR 2再进化,对图像理解已经像人一样逻辑推理了

25年10月DeepSeek-OCR发布,用视觉大幅压缩文本,“1张图片装10页书的信息!DeepSeek-OCR用图片压缩文本,一图胜千言被实现了”,震惊了整个科技圈。 刚刚,DeepSeek-OCR 2再升级,保持了前代DeepSee…

作者头像 李华
网站建设 2026/8/25 16:45:18

Coherent 与 Quside 演示基于 VCSEL 的熵源

专注于大规模制造解决方案,以支持广泛应用场景下下一代安全架构的光子学公司 Coherent 与量子技术公司 Quside 共同宣称,他们于基于硬件的安全领域斩获一项重大里程碑式成果——成功演示了一种具备大规模制造潜力的量子熵源。此解决方案依托 Coherent 的…

作者头像 李华
网站建设 2026/9/2 21:23:48

深度测评!本科生论文痛点TOP9 AI论文软件全解析

深度测评!本科生论文痛点TOP9 AI论文软件全解析 2026年本科生论文写作工具测评:为何需要一份权威榜单? 随着AI技术的不断进步,越来越多的本科生开始依赖AI论文软件来提升写作效率、优化内容结构。然而,面对市场上五花八…

作者头像 李华
网站建设 2026/9/2 22:53:06

Oracle迁移KingbaseES:从痛点到破局之道

引言 在国产化替代浪潮下,Oracle数据库迁移已成为众多企业的必答题。然而,这场看似简单的"搬家"行动,实则暗藏重重挑战。本文将从实际迁移场景出发,深入剖析Oracle迁移至电科金仓KingbaseES过程中的核心痛点,并探讨金仓如何通过技术创新破解这些难题。 一、迁移背景…

作者头像 李华
网站建设 2026/9/2 21:23:36

重庆思庄oracle技术分享-Oracle 快速生成 AWR 报告

适用场景: -性能问题快速诊断 -手动触发 AWR 快照收集 -生成指定时段的性能分析报告 操作步骤: 1. 登录数据库并创建手动快照 sql -- 以 SYSDBA 身份登录(可根据环境选择登录方式) CONNECT / AS SYSDBA; -- 创建手动 AWR 快照…

作者头像 李华