PyTorch

PyTorch 是一个 Python 优先的深度学习框架，也是使用 GPU 和 CPU 优化的深度学习张量库，能够在强大的 GPU 加速基础上实现张量和动态神经网络。

一. 命令行提交

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

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

mkdir pytorchJob1
cd pytorchJob1

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

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

#!/bin/bash
module add Anaconda3            #加载Anaconda3软件
source activate pytorch-1.8.1   #激活pytorch环境
python mnist.py > py.log        #运行程序

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

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

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

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

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

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