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:
Pripremljen je modul koji postavlja sve potrebne varijable okoline:
Verzija | Modul |
---|---|
1.0.0 | pytorch/1-0-0-gpu |
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.
U opisima poslova koji koriste jedan grafički procesor, obavezno treba koristiti cuda-wrapper.sh, kao u sljedećem primjeru: |
#$ -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 s bibliotekom PyTorch trenutno nije moguće. |
Posao se podnosi s pristupnog čvora naredbom:
qsub pytorch.sge |