并行 云架构 深度框架 sbatch slurm 深度学习 tensorflow环境从搭建到使用 conda

本文主要是介绍并行 云架构 深度框架 sbatch slurm 深度学习 tensorflow环境从搭建到使用 conda,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

有一定的GPU云时常可用,一个节点4个GPU,我本人决定使用anaconda搭建tensorflow1.13并且使用。

anaconda是乙方提供的,使用bash命令可以加载

module load anaconda/3.7

加载后正常使用create命令建立环境

详情见我所有conda标志的博客,其实就是下面一句代码,看明白就不用翻了。

 下面这句代码就从零开始建立了一个tensorflow gpu的环境,版本是1.13,这一句代码,所有的依赖包括cuda,cudnn都保证安装好了。并且不会和任何人包括自己的环境冲突,虚拟环境这个技能真的超级有用且简单。不明白为什么很多人非要看十多篇长达5页的博客,折腾半个月环境,偏偏不愿意花五分钟试试我下面的代码。

conda create --name tenf13 tensorflow-gpu=1.13

上面建立的虚拟环境名字叫做tenf13,名字就是你叫他他就答应的。所以需要点名激活他,你不激活用不了,用完了,你再让他走。虚拟环境就是随叫随到这种的方便,所以可以建立10多种,随便是pytorch,tf,keras,还有各种版本,比如tf1,tf2。

激活环境

注意,必须使用source激活该环境。

source activate tenf13

这个环境就搭建完成了。使用时是这么使用的。

在你的bash代码中。我的实验配置都是用bash代码写得,因为方便。

#!/bin/bash#SBATCH -N 2
#SBATCH --ntasks-per-node=20
#SBATCH -A para
#SBATCH -p gpu
#SBATCH --gres=gpu:4export HOME=/home/tom/project
module load anaconda/3.7
source activate tenf13

实际上就最后两句有用的激活了环境。export HOME这句我觉着可能是定位anaconda的。这两句激活了环境,下面就可以写自己的代码了。比如

#!/bin/bash#SBATCH -N 2
#SBATCH --ntasks-per-node=20
#SBATCH -A para
#SBATCH -p gpu
#SBATCH --gres=gpu:4export HOME=/home/tom/project
module load anaconda/3.7
source activate tenf13python test.py

bash 代码也没什么神秘的,就是平时输入命令行的现在输入在一个文件里面而已。

tensorflow代码

之所以写这个是因为,在配置session的config时,需要特别注意一个参数。

否则会出错:CUB segmented reduce errortoo many resources requested for launch

参考:

https://devtalk.nvidia.com/default/topic/1038115/jetson-tx2/cub-segmented-reduce-errortoo-many-resources-requested-for-launch/

config = tf.ConfigProto()
config.gpu_options.allow_growth = Truesession = tf.Session(config=config, ...)

所以在我的实际代码中我是这么写的:

      # Soft placement allows placing on CPU ops without GPU implementation.session_config = tf.ConfigProto(allow_soft_placement=True, log_device_placement=False)#LT add on cloud envsession_config.gpu_options.allow_growth = True

运行与监控

使用sbatch命令可以运行.sh文件。提交的进程会有一个唯一的id,比如会这么回复:

Submitted batch job 6712625

使用squeue可以查看正在运行的id

输出的内容在

slurm-6712625.out

使用vim即可看

 

 

这篇关于并行 云架构 深度框架 sbatch slurm 深度学习 tensorflow环境从搭建到使用 conda的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1128400

相关文章

Python并行处理实战之如何使用ProcessPoolExecutor加速计算

《Python并行处理实战之如何使用ProcessPoolExecutor加速计算》Python提供了多种并行处理的方式,其中concurrent.futures模块的ProcessPoolExecu... 目录简介完整代码示例代码解释1. 导入必要的模块2. 定义处理函数3. 主函数4. 生成数字列表5.

Python中help()和dir()函数的使用

《Python中help()和dir()函数的使用》我们经常需要查看某个对象(如模块、类、函数等)的属性和方法,Python提供了两个内置函数help()和dir(),它们可以帮助我们快速了解代... 目录1. 引言2. help() 函数2.1 作用2.2 使用方法2.3 示例(1) 查看内置函数的帮助(

Python虚拟环境与Conda使用指南分享

《Python虚拟环境与Conda使用指南分享》:本文主要介绍Python虚拟环境与Conda使用指南,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录一、python 虚拟环境概述1.1 什么是虚拟环境1.2 为什么需要虚拟环境二、Python 内置的虚拟环境工具

Linux脚本(shell)的使用方式

《Linux脚本(shell)的使用方式》:本文主要介绍Linux脚本(shell)的使用方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录概述语法详解数学运算表达式Shell变量变量分类环境变量Shell内部变量自定义变量:定义、赋值自定义变量:引用、修改、删

Java使用HttpClient实现图片下载与本地保存功能

《Java使用HttpClient实现图片下载与本地保存功能》在当今数字化时代,网络资源的获取与处理已成为软件开发中的常见需求,其中,图片作为网络上最常见的资源之一,其下载与保存功能在许多应用场景中都... 目录引言一、Apache HttpClient简介二、技术栈与环境准备三、实现图片下载与保存功能1.

Python中使用uv创建环境及原理举例详解

《Python中使用uv创建环境及原理举例详解》uv是Astral团队开发的高性能Python工具,整合包管理、虚拟环境、Python版本控制等功能,:本文主要介绍Python中使用uv创建环境及... 目录一、uv工具简介核心特点:二、安装uv1. 通过pip安装2. 通过脚本安装验证安装:配置镜像源(可

LiteFlow轻量级工作流引擎使用示例详解

《LiteFlow轻量级工作流引擎使用示例详解》:本文主要介绍LiteFlow是一个灵活、简洁且轻量的工作流引擎,适合用于中小型项目和微服务架构中的流程编排,本文给大家介绍LiteFlow轻量级工... 目录1. LiteFlow 主要特点2. 工作流定义方式3. LiteFlow 流程示例4. LiteF

使用Python开发一个现代化屏幕取色器

《使用Python开发一个现代化屏幕取色器》在UI设计、网页开发等场景中,颜色拾取是高频需求,:本文主要介绍如何使用Python开发一个现代化屏幕取色器,有需要的小伙伴可以参考一下... 目录一、项目概述二、核心功能解析2.1 实时颜色追踪2.2 智能颜色显示三、效果展示四、实现步骤详解4.1 环境配置4.

使用jenv工具管理多个JDK版本的方法步骤

《使用jenv工具管理多个JDK版本的方法步骤》jenv是一个开源的Java环境管理工具,旨在帮助开发者在同一台机器上轻松管理和切换多个Java版本,:本文主要介绍使用jenv工具管理多个JD... 目录一、jenv到底是干啥的?二、jenv的核心功能(一)管理多个Java版本(二)支持插件扩展(三)环境隔

SQL中JOIN操作的条件使用总结与实践

《SQL中JOIN操作的条件使用总结与实践》在SQL查询中,JOIN操作是多表关联的核心工具,本文将从原理,场景和最佳实践三个方面总结JOIN条件的使用规则,希望可以帮助开发者精准控制查询逻辑... 目录一、ON与WHERE的本质区别二、场景化条件使用规则三、最佳实践建议1.优先使用ON条件2.WHERE用