fix reuse local and trial (#3408)

nni/tools/trial_tool/trial.py

@@ -3,6 +3,7 @@
 
 import ctypes
 import os
+import sys
 import shlex
 import tarfile
 import time
@@ -88,9 +89,12 @@ class Trial:
 
         trial_command = self.args.trial_command
 
-        gpuIndices = self.data.get('gpuIndices')
+        gpuIndices = self.data.get("gpuIndices")
         if (gpuIndices is not None):
-            trial_command = 'CUDA_VISIBLE_DEVICES="%s " %s' % (gpuIndices, trial_command)
+            if sys.platform == "win32":
+                trial_command = 'set CUDA_VISIBLE_DEVICES="%s " && call %s' % (gpuIndices, trial_command)
+            else:
+                trial_command = 'CUDA_VISIBLE_DEVICES="%s " %s' % (gpuIndices, trial_command)
 
         self.log_pipe_stdout = self.trial_syslogger_stdout.get_pipelog_reader()
         self.process = Popen(trial_command, shell=True, stdout=self.log_pipe_stdout,

ts/nni_manager/training_service/reusable/environments/localEnvironmentService.ts

@@ -92,9 +92,11 @@ export class LocalEnvironmentService extends EnvironmentService {
     private getScript(environment: EnvironmentInformation): string[] {
         const script: string[] = [];
         if (process.platform === 'win32') {
+            script.push(`$env:PATH="${process.env.path}"`)
             script.push(`cd $env:${this.experimentRootDir}`);
             script.push(`New-Item -ItemType "directory" -Path ${path.join(this.experimentRootDir, 'envs', environment.id)} -Force`);
-            environment.command = `cd envs\\${environment.id} && python -m nni.tools.trial_tool.trial_runner`;
+            script.push(`cd envs\\${environment.id}`);
+            environment.command = `python -m nni.tools.trial_tool.trial_runner`;
             script.push(
                 `cmd.exe /c ${environment.command} --job_pid_file ${path.join(environment.runnerWorkingFolder, 'pid')} 2>&1 | Out-File "${path.join(environment.runnerWorkingFolder, 'trial_runner.log')}" -encoding utf8`,
                 `$NOW_DATE = [int64](([datetime]::UtcNow)-(get-date "1/1/1970")).TotalSeconds`,