robin-courant commited on
Commit
7d9ffec
1 Parent(s): 66da254

Create gpu.yaml

Browse files
Files changed (1) hide show
  1. configs/compnode/gpu.yaml +3 -0
configs/compnode/gpu.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ device: cuda
2
+ num_gpus: 1
3
+ num_workers: 8