PyTorch 1.0.0

Na računalnom klasteru Isabella, na čvorovima s grafičkim procesorima NVIDIA Tesla V100-SXM2-16GB, instalirana je Pythonova biblioteka za strojno učenje PyTorch, verzija 1.0.0.

PyTorch je instaliran za Python 3.5, uz sljedeće verzije NVIDIA alata i biblioteka za strojno učenje:

• CUDA 10.0
• cuDNN 7.3.1
• NCCL 2.3.5

Pripremljen je modul koji postavlja sve potrebne varijable okoline:

Izvođenje poslova

U nastavku je opisano podnošenje tipičnih Python poslova. Više informacija o pokretanju poslova možete pronaći na stranicama Korištenje grafičkih procesora te Pokretanje i upravljanje poslovima.

Korištenje jednog grafičkog procesora:

#$ -cwd
#$ -pe gpu 1
 
module load pytorch/1-0-0-gpu
 
cuda-wrapper.sh python3.5 moj_program.py

Korištenje više grafičkih procesora:

Korištenje više grafičkih procesora na jednom čvoru:

Za pokretanje poslova koji zahtijevaju više grafičkih procesora na jednom čvoru potrebno je koristiti paralelnu okolinu gpusingle te željeni broj grafičkih procesora (maksimalno 4):

#$ -cwd
#$ -pe gpusingle 4
 
module load pytorch/1-0-0-gpu
 
openmpi-wrapper.sh python3.5 moj_program.py

Korištenje više grafičkih procesora na više čvorova:

#$ -cwd
#$ -pe gpu 6
 
module load pytorch/1-0-0-gpu
 
openmpi-wrapper.sh python3.5 moj_program.py

Korištenje cijelih čvorova s grafičkim procesorima

Za pokretanje poslova koji zahtijevaju cijele čvorove potrebno je koristiti paralelnu okolinu gpufull te željeni broj grafičkih procesora (mora biti djelitelj broja 4). Primjer skripte za zauzimanje dva čvora:

#$ -cwd
#$ -pe gpufull 8
 
module load pytorch/1-0-0-gpu
 
openmpi-wrapper.sh python3.5 moj_program.py

Podnošenje posla

Posao se podnosi s pristupnog čvora naredbom:

qsub pytorch.sge