20 lines
609 B
Bash
20 lines
609 B
Bash
|
#!/bin/bash
|
||
|
|
#SBATCH --job-name=job_name
|
||
|
|
#SBATCH --time=10:00:00
|
||
|
|
#SBATCH --account= ### to fill
|
||
|
|
#SBATCH --partition=plgrid-gpu-v100
|
||
|
|
#SBATCH --cpus-per-task=1
|
||
|
|
#SBATCH --ntasks-per-node=1
|
||
|
|
#SBATCH --gres=gpu:1
|
||
|
|
|
||
|
|
source path/to/mambaforge/bin/activate ### to change
|
||
|
|
conda activate epos-ai-train
|
||
|
|
|
||
|
|
|
||
|
|
python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
|
||
|
|
python -c "import torch; print('Number of CUDA devices:', torch.cuda.device_count())"
|
||
|
|
python -c "import torch; print('Name of GPU:', torch.cuda.get_device_name(torch.cuda.current_device()))"
|
||
|
|
|
||
|
|
|
||
|
|
python pipeline.py --dataset "bogdanka"
|