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
This commit is contained in:
allegroai
2019-11-08 22:36:24 +02:00
parent 4a8f52b5a5
commit a7873705ec
6 changed files with 81 additions and 22 deletions

View File

@@ -30,7 +30,7 @@ class VirtualenvPip(SystemPip, PackageManager):
self.session = session
self.path = path
self.requirements_manager = requirements_manager
self.python = "python{}".format(python)
self.python = python
def _make_command(self, command):
return self.session.command(self.bin, "-m", "pip", *command)