Skip to main content

TensorFlow

TensorFlow 是一个端到端开源机器学习平台。它拥有一个包含各种工具、库和社区资源的全面灵活生态系统，可以让研究人员推动机器学习领域的先进技术的发展，并让开发者轻松地构建和部署由机器学习提供支持的应用。

一. 命令行提交​

通过SSH连接创建并连接管理节点。

Step 1. 创建作业目录并进入；

mkdir tensorflowJob1
cd tensorflowJob1

Step 2. 通过文件传输上传输入文件tf.py，详情请查看Linux数据传输；

Step 3. 在该文件夹下创建如下执行脚本tensorflow.sh：

#!/bin/bash
module add Anaconda3/2020.02          #加载Anaconda
source activate tensorflow-gpu-2.2    #激活tensorflow环境
python cloudam.py > tf.log            #运行程序

Step 4. 使用sbatch命令提交作业；

提交任务到带有一张T4卡的GPU节点运行。

sbatch -p g-t4-1 -c 4 tensorflow.sh

查看作业运行情况及参数详细介绍请点击查看slurm命令。

结果文件下载请查看Linux数据传输。

点击下载以上作业样例：TensorFlow.tar.gz