wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

WIP

This commit is contained in:
Remi Cadene 2024-03-04 17:32:29 +00:00
parent 2bcf2631b9
commit 20c08bb740
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
sbatch.sh
View File

@ -1,22 +1,24 @@
#!/bin/bash #!/bin/bash
#SBATCH --nodes=1 # total number of nodes (N to be defined) #SBATCH --nodes=1 # total number of nodes (N to be defined)
#SBATCH --ntasks-per-node=1 # number of tasks per node (here 8 tasks, or 1 task per GPU) #SBATCH --ntasks-per-node=1 # number of tasks per node (here 8 tasks, or 1 task per GPU)
#SBATCH --qos=normal # number of GPUs reserved per node (here 8, or all the GPUs)
#SBATCH --partition=hopper-prod
#SBATCH --gres=gpu:1 # number of GPUs reserved per node (here 8, or all the GPUs) #SBATCH --gres=gpu:1 # number of GPUs reserved per node (here 8, or all the GPUs)
#SBATCH --cpus-per-task=8 # number of cores per task (8x8 = 64 cores, or all the cores) #SBATCH --cpus-per-task=12 # number of cores per task
#SBATCH --time=2-00:00:00 #SBATCH --mem-per-cpu=11G
#SBATCH --output=/home/rcadene/slurm/%j.out #SBATCH --time=12:00:00
#SBATCH --error=/home/rcadene/slurm/%j.err #SBATCH --output=/admin/home/remi_cadene/slurm/%j.out
#SBATCH --qos=medium #SBATCH --error=/admin/home/remi_cadene/slurm/%j.err
#SBATCH --mail-user=re.cadene@gmail.com #SBATCH --mail-user=remi_cadene@huggingface.co
#SBATCH --mail-type=ALL #SBATCH --mail-type=ALL
CMD=$@ CMD=$@
echo "command: $CMD" echo "command: $CMD"
apptainer exec --nv \ # apptainer exec --nv \
~/apptainer/nvidia_cuda:12.2.2-devel-ubuntu22.04.sif $SHELL # ~/apptainer/nvidia_cuda:12.2.2-devel-ubuntu22.04.sif $SHELL
source ~/.bashrc #source ~/.bashrc
conda activate fowm #conda activate fowm
srun $CMD srun $CMD