Merge pull request deepmodeling#302 from AnguseZhang/devel

Fix bug if cuda_multi_tasks is used
njzjz · Jul 26, 2020 · 0765b1c · 0765b1c
2 parents 6c4462d + 4759912
commit 0765b1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dpgen/dispatcher/Batch.py b/dpgen/dispatcher/Batch.py
@@ -155,7 +155,7 @@ def _sub_script_inner(self,
             else :
                 # do not support task-wise restart
                 tmp_cmd = ' %s 1>> %s 2>> %s ' % (self.sub_script_cmd(cmd, jj, res), outlog, errlog)
-                ret += 'CUDA_VISIBLE_DEVICES=%d %s &\n\n' % ((self.cmd_cnt % self.manual_cuda_devices), tmp_cmd)
+                ret += 'export CUDA_VISIBLE_DEVICES=%d; %s &\n\n' % ((self.cmd_cnt % self.manual_cuda_devices), tmp_cmd)
                 self.cmd_cnt += 1
             ret += 'cd %s\n' % self.context.remote_root
             ret += 'test $? -ne 0 && exit 1\n'