search

BayesWatch / pytorch-experiments-template

A pytorch based classification experiments template
GNU General Public License v3.0
5 stars 1 forks source link

Automating GPU allocation #16

Closed AntreasAntoniou closed 3 years ago

AntreasAntoniou commented 3 years ago

This is also something I have ready to go. I use GPUtil to find available GPUs and allocate to a job depending on how many GPUs you require. If GPUs are not available, the script will let you know, and it'll just keep checking and waiting until what you need becomes available.

AntreasAntoniou commented 3 years ago

This has been implemented.