Applications/Scheduler

[slurm] rpmbuild with gpu

CRAY KOREA Blog 2023. 11. 14. 15:25

※ slurm gres.conf 사용을 위한 간단한 예제

 

- cuda toolkit 설치

$ wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
$ sudo sh cuda_11.8.0_520.61.05_linux.run

 

- rpmbuild 옵션에 "--with-nvml"을 추가

$ rpmbuild --define "_with_nvml --with-nvml=/usr/local/cuda-11.8" -ta slurm-22.05.6.tar.bz2

 

- gpu 라이브러리 포함 확인

$ cd ${HOME}/rpmbuild/RPMS/x86_64
$ rpm -qlp slurm-22.05.6-1.el8.x86_64.rpm | grep -i nvml
/usr/lib64/slurm/gpu_nvml.so