news 2026/2/17 0:34:57

深度学习 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/2/4 17:31:26

农业数据平台如何利用KindEditor实现WORD图表复制?

Word一键转存CMS升级日记 📝 Day 1:需求分析与市场调研 作为一个"资深"的广东计科大三狗🐶,今天开始给我的CMS新闻系统升级Word粘贴功能!老板说"预算99块不能再多"(笑死,…

作者头像 李华
网站建设 2026/2/10 13:17:15

vue.js中WebUploader在局域网上传大文件如何分块?

大三学弟的大文件上传救星:原生JSPython全栈方案(附完整前端代码) 兄弟,作为刚啃完《计算机网络》课本、正对着VS Code发懵的网工大三学弟,我太懂你现在的处境了——老师要大文件上传的毕设作品,网上开源代…

作者头像 李华
网站建设 2026/2/15 0:12:52

互联网大厂Java求职面试实战:微服务与AI全栈技术问答解析

互联网大厂Java求职面试实战:微服务与AI全栈技术问答解析 本文以一个严肃的面试官与搞笑水货程序员谢飞机的对话形式,展现互联网大厂Java求职者面试全过程。涵盖Java SE 8/11/17、Jakarta EE、Spring Boot、微服务架构、数据库ORM、缓存、消息队列、安全…

作者头像 李华
网站建设 2026/2/11 18:07:18

军工项目中使用KindEditor如何安全导入WORD涉密图片?

政务信息化解决方案:Word/PPT/Excel/PDF全格式集成方案(信创兼容版) 一、方案概述(政务场景定制化) 针对党政机关、事业单位对国产化兼容、数据安全、多格式集成的核心需求,我们推出**「智文通」政务文档集…

作者头像 李华
网站建设 2026/2/16 6:14:29

能源化工SpringMVC如何上传大文件报告?

大文件传输系统建设方案 一、项目背景与需求分析 作为山西某大型国企上市公司的项目负责人,我司目前承担着集团级大文件传输平台建设任务。客户群体涵盖部委、政府部门及大型央企,对系统安全性、稳定性及兼容性提出严苛要求。经详细调研,核…

作者头像 李华
网站建设 2026/2/15 11:29:02

企业视频培训平台如何实现授权观看和用户分析?

摘要:企业可通过视频培训专栏快速构建内部培训平台,实现授权观看和用户分析。操作步骤:1)新建专栏并设置观看权限;2)添加视频内容;3)批量导入和管理用户;4)通…

作者头像 李华