Commit Graph

6 Commits

Author SHA1 Message Date
allegroai
a7873705ec Add --gpus / --cpu-only (equivalent to NVIDIA_VISIBLE_DEVICE)
Add agent.python_binary specifying full path to python binary to use for virtual environement creation
Fix Windows support
2019-11-08 22:36:24 +02:00
allegroai
c352c2711c Improve multiple GPU's docker PyTorch support
Fix potential zombie dockers if task is aborted
2019-11-01 20:24:53 +02:00
allegroai
1765788e80 Improve PyTorch CUDA auto package resolving 2019-10-28 21:54:11 +02:00
allegroai
5e6a809efd Remove deprecated command 2019-10-28 21:53:34 +02:00
allegroai
d2c5e127b8 Add check for trains_agent updates 2019-10-28 21:52:55 +02:00
allegroai
224a709f40 Initial release 2019-10-25 22:28:44 +03:00