...
Code Block |
---|
username@lanta:~> apptainer pull tensorflow-_2.3.1-gpu.sif docker://tensorflow/tensorflow:2.3.1-gpu |
...
ไฟล์ Setup.py - ใช้สำหรับการดาวน์โหลด Data set
ไฟล์ MNIST.py - ใช้สำหรับทดสอบรัน Training
ไฟล์ submit.sh - ใช้สำหรับส่ง Job รันบนเครื่อง GPU
...
Code Block |
---|
#!/bin/bash #SBATCH -p gpu # Specify partition [Compute/Memory/GPU] #SBATCH -N 1 -c 16 # Specify number of nodes and processors per task #SBATCH --gpus-per-node=4 # Specify number of GPU per task #SBATCH --ntasks-per-node=4 # Specify tasks per node #SBATCH -t 120:00:00 # Specify maximum time limit (hour: minute: second) #SBATCH -A ltxxxxxx # Specify project name #SBATCH -J JOBNAME # Specify job name module load Apptainer/1.1.6 # Load the Apptainer module apptainer exec --nv -B $PWD:$PWD filetensorflow_2.3.1-gpu.sif python3 fileMNIST.py # Run your program |
Info |
---|
|
การส่ง Job เข้ารันบน LANTA
ดาวน์โหลด Data set โดยใช้คำสั่งต่อไปนี้
Code Block |
---|
ml Apptainer
apptainer exec tensorflow_2.3.1-gpu.sif python Setup.py |
จากนั้น ใช้คำสั่ง
sbatch submit.sh
เพื่อส่ง Job ของคุณเข้าระบบ Slurm ของ LANTA
Code Block |
---|
sbatch submit.sh |