资源申请
1、申请CPU(数字对应内存)
salloc(例申请到cpu3)
ssh cpu3
2、释放资源
scancel 资源号
创建虚拟环境
1、创建虚拟环境
conda create --name 环境名称 python==3.8
2、激活虚拟环境
source activate 环境名称
3、退出虚拟环境
source deactivate
4、查看conda环境下所有的虚拟环境
conda env list
安装必要包
1、安装pytorch
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
2、 安装必要的包(例matplotlib)
pip install matplotlib
上传作业
1、代码文件拖入MobaXterm
2、进入文件目录
cd test
(退出)cd ..
3、执行文件里写的sbatch
sbatch test.txt
进root改环境变量
vim ~/.bashrc
退出
source ~/.bashrc
解决办法:
先dos2unix MI_v3_verify.txt
然后sbatch MI_v3_verify.txt
不能用sbatch执行时,用srun
srun -p gpu-a40 --gres=gpu:1 加上python attack.py等
创建会话挂后台
(退出时ctrl B + D)
脚本格式
#!/bin/sh
#SBATCH -J test
#SBATCH -o log/test.out.%j
#SBATCH -e log/test.err.%j
#SBATCH --partition=gpu-a10
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --ntasks-per-node=1
#SBATCH --gres=gpu:1
python test.py --param1 value1 --param2 value2
申请GPU
salloc -N 1 -n 1 --gres=gpu:1 -p gpu-a10