bdh-generalization / requirements

1 stars 0 forks source link

GPU support for vantage6 node containers #3

Open AnanyaCN opened 1 year ago

AnanyaCN commented 1 year ago

Deep learning studies need access to GPU devices in the data nodes. We have implemented it at the node level for a separate version https://github.com/AnanyaCN/argos-node/blob/master/vantage6/node/docker_manager.py #362. This needs to be tested accross devices and and made more generic.

bartvanb commented 1 year ago

Relevant: https://github.com/vantage6/vantage6/issues/597