diff --git a/bob/bio/base/config/grid/gpu.py b/bob/bio/base/config/grid/gpu.py new file mode 100644 index 0000000000000000000000000000000000000000..ecc6e475139e7ebc854d924e6415f4b708f74343 --- /dev/null +++ b/bob/bio/base/config/grid/gpu.py @@ -0,0 +1,16 @@ +import bob.bio.base + +# define a queue with demanding parameters +grid = bob.bio.base.grid.Grid( + training_queue = 'GPU', + # preprocessing + preprocessing_queue = '4G', + # feature extraction + extraction_queue = 'GPU', + # feature projection + projection_queue = '4G', + # model enrollment + enrollment_queue = '4G', + # scoring + scoring_queue = '4G' +) diff --git a/bob/bio/base/grid.py b/bob/bio/base/grid.py index f11fbd3bc0868554c2b6b30594b60ad1f4bd2ae3..447575debbbe7246926f7219e4112285cf26040e 100644 --- a/bob/bio/base/grid.py +++ b/bob/bio/base/grid.py @@ -29,7 +29,8 @@ PREDEFINED_QUEUES = { '16G-io-big' : {'queue' : 'q1dm', 'memfree' : '16G', 'pe_opt' : 'pe_mth 2', 'hvmem' : '8G', 'io_big' : True}, '32G' : {'queue' : 'q1dm', 'memfree' : '32G', 'pe_opt' : 'pe_mth 4', 'hvmem' : '8G', 'io_big' : True}, '64G' : {'queue' : 'q1dm', 'memfree' : '64G', 'pe_opt' : 'pe_mth 8', 'hvmem' : '8G', 'io_big' : True}, - 'Week' : {'queue' : 'q1wm', 'memfree' : '32G', 'pe_opt' : 'pe_mth 4', 'hvmem' : '8G'} + 'Week' : {'queue' : 'q1wm', 'memfree' : '32G', 'pe_opt' : 'pe_mth 4', 'hvmem' : '8G'}, + 'GPU' : {'queue' : 'gpu'} } class Grid: diff --git a/setup.py b/setup.py index 25e288b4a4ecc6b5826b577ec9847a74c79435ae..5d0650d705fa6ee6d969ce31c1d889e3632ba8c0 100644 --- a/setup.py +++ b/setup.py @@ -145,6 +145,7 @@ setup( 'local-p16 = bob.bio.base.config.grid.local:grid_p16', 'grid = bob.bio.base.config.grid.grid:grid', 'demanding = bob.bio.base.config.grid.demanding:grid', + 'gpu = bob.bio.base.config.grid.gpu:grid', ], },