hpc:gpu
Skirtumai
Čia matote skirtumus tarp pasirinktos versijos ir esamo dokumento.
Next revision | Previous revisionNext revisionBoth sides next revision | ||
hpc:gpu [2021/12/16 10:57] – sukurtas eduardas | hpc:gpu [2022/01/31 16:46] – linp | ||
---|---|---|---|
Linija 1: | Linija 1: | ||
- | Info apie GPU skaičiavimus | + | ====== Paketinis užduočių vykdymas su GPU (SLURM) ====== |
+ | |||
+ | Norint pasinaudoti PST skaičiavimo resursais su GPU reikės formuoti užduočių skriptus (.sh): | ||
+ | |||
+ | |||
+ | Pavyzdys, bandant patikrinti ar jum suteikta GPU. | ||
+ | Sukuriamas failas gpu.txt su indikatoriumi ir vykdymo laiko momentu: | ||
+ | |||
+ | <code shell test_gpu.sh> | ||
+ | #!/bin/sh | ||
+ | #SBATCH -p gpu | ||
+ | #SBATCH -n1 | ||
+ | #SBATCH --gres gpu | ||
+ | |||
+ | source gpu_env/ | ||
+ | python3 test.py | ||
+ | </ | ||
+ | |||
+ | Pasinagrinėkime pavyzdį paeilučiui: | ||
+ | <code shell> | ||
+ | $ #SBATCH -p gpu | ||
+ | </ | ||
+ | nukreipiama į PST su vaizdo plokščių resursais skaičiavimo resursą. | ||
+ | |||
+ | <code shell> | ||
+ | $ #SBATCH -n1 | ||
+ | </ | ||
+ | nurodome, koks CPU poreikis bus reikalingas | ||
+ | |||
+ | <code shell> | ||
+ | $ #SBATCH --gres gpu | ||
+ | </ | ||
+ | nurodome, koks GPU poreikis bus reikalingas (esant N GPU poreikiui būtų nurodoma gpu:N). | ||
+ | |||
+ | |||
+ | <code shell> | ||
+ | $ source gpu_env/ | ||
+ | </ | ||
+ | pasiruošiame aplinką darbui (pirmą kartą ją reikia sukurti atskirai žr. aplinka.sh) | ||
+ | |||
+ | <code shell> | ||
+ | $ python3 test.py | ||
+ | </ | ||
+ | Kviečiame savo kodo skriptą, atlikti užduočiai su Python | ||
+ | |||
+ | |||
+ | |||
+ | Čia iškviečiamo Python kalbos skripto kodas, iškviečiant Pytorch biblioteką ir patikrinat GPU prieigą, kurios dažniausiai reikia kuriant mašininio mokymo modelius. | ||
+ | |||
+ | <code shell test.py> | ||
+ | import torch | ||
+ | import time | ||
+ | |||
+ | val = torch.cuda.is_available() | ||
+ | |||
+ | f = open(" | ||
+ | ts = time.time() | ||
+ | f.write(" | ||
+ | f.close() | ||
+ | </ | ||
+ | |||
+ | |||
+ | |||
+ | Čia iškviečiamo Python kalbos skripto kodas, iškviečiant Pytorch biblioteką ir patikrinat GPU prieigą, kurios dažniausiai reikia kuriant mašininio mokymo modelius. | ||
+ | |||
+ | <code shell aplinka.sh> | ||
+ | # | ||
+ | |||
+ | python3 -m venv gpu_env | ||
+ | |||
+ | # pirmą kartą susirašome reikiamas bibliotekas savo projektui pvz: | ||
+ | |||
+ | pip3 install wheel | ||
+ | pip3 install pillow scikit-image | ||
+ | pip3 install numba | ||
+ | pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 -f https:// | ||
+ | |||
+ | </ |
hpc/gpu.txt · Keista: 2022/10/30 13:54 vartotojo linp