mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-15 02:05:37 +00:00
14 lines
167 B
Bash
14 lines
167 B
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
lrank=$SLURM_LOCALID
|
||
|
|
lgpu=(0 1 2 3 7 6 5 4)
|
||
|
|
|
||
|
|
export ROCR_VISIBLE_DEVICES=${lgpu[$lrank]}
|
||
|
|
|
||
|
|
echo "`hostname` - $lrank device=$ROCR_VISIBLE_DEVICES "
|
||
|
|
|
||
|
|
$*
|
||
|
|
|
||
|
|
|
||
|
|
|