Versions Compared

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.

...

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.

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

  • CUDA
  • cuDNN 7.3.1
  • NCCL 2.3.5

Pripremljen je modul koji postavlja sve potrebne varijable okoline:

VerzijaCUDAModul
1.0.010.0pytorch/1-0-0-gpu
1.3.010.0pytorch/1.3.0-gpu
1.5.010.1pytorch/1.5.0-gpu
1.5.110.1pytorch/1.5.1-gpu
1.8.111pytorch/1.8.1-gpu

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.

...

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

Korištenje više grafičkih procesora:

...