Arvados

Feature #12630

Updated by Peter Amstutz about 7 years ago

<pre> 
 $namespaces: 
   nvidia: https://www.nvidia.com/object/ 
 requirements: 
   - class: arv:HardwareRequirement 
     hardware: 
       - hwtype: nvidia:tesla-k80 
         hwcount: 1 
 </pre> 

 In container_request: 

 <pre> 
 runtime_constraints: 
   GPU: 
     - driver: nvidia 
       type: tesla-k80 
       count: 1 
 </pre> 

 Scheduled as: 

 <pre> 
 sbatch --gres=gpu:tesla-k80:1 
 </pre> 

 Crunch-run, setting up container, when RuntimeConstraints.GPU.Driver == nvidia: 

 * Inject devices /dev/nvidiactl, /dev/nvidia-uvm and /dev/nvidia[0-999] into container 
 * Inject environment variable CUDA_VISIBLE_DEVICES 

Back