news 2026/2/28 19:46:21

深度学习 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/27 4:01:37

vue3如何通过WebUploader在局域网实现大文件断点续传?

前端程序员外包项目救星:原生JS大文件上传组件(Vue3实现) 兄弟,作为在杭州接外包的老前端程序员,太懂你现在的处境了——甲方要20G大文件上传,还要兼容IE9,预算卡得死死的,网上代码…

作者头像 李华
网站建设 2026/2/24 2:05:33

学长亲荐!专科生必备!2026 TOP10 AI论文网站深度测评

学长亲荐!专科生必备!2026 TOP10 AI论文网站深度测评 为什么需要一份专为专科生定制的AI论文网站榜单? 随着人工智能技术的不断进步,AI写作工具已经成为学术研究中不可或缺的辅助工具。对于专科生而言,撰写论文不仅是…

作者头像 李华
网站建设 2026/2/27 14:57:35

ArcGIS赋能水文水环境保护:从基础操作到高级分析,涵盖数据库构建、空间插值、水文模拟与水环境容量计算的综合技能

水文水环境保护是我国生态文明建设的重要组成部分,其涉及的数据具有显著的空间分布特征,传统方法难以高效处理与分析。地理信息系统(GIS)凭借其强大的空间数据管理、编辑和分析能力,成为解决水文水环境空间差异问题的关…

作者头像 李华
网站建设 2026/2/26 20:54:33

PowerAmerica 宣布举办宽禁带(WBG)短期课程

1. 课程概况 (Course Overview)形式:为期三天的在线课程(通过 Zoom 进行)。时间:2026年2月3日 - 2月5日(为期3天)。主办方:PowerAmerica(美国能源部支持的机构,致力于连接…

作者头像 李华
网站建设 2026/2/16 23:00:50

无人机视角城市街道行人与车辆检测数据集VOC+YOLO格式5291张2类别

数据集格式:Pascal VOC格式YOLO格式(不包含分割路径的txt文件,仅仅包含jpg图片以及对应的VOC格式xml文件和yolo格式txt文件) 图片数量(jpg文件个数):5291 标注数量(xml文件个数):5291 标注数量(txt文件个数):5291 …

作者头像 李华