Просмотр исходного кода

change order of gpu 0 and 1

master
Jeong Geol Kim 4 лет назад
Родитель
Сommit
60d85ffe66
1 измененных файлов: 2 добавлений и 2 удалений
  1. 2
    2
      auto_extract.py

+ 2
- 2
auto_extract.py Просмотреть файл

@@ -35,11 +35,11 @@ def main():
35 35
                     r_time = time.ctime(time.time())
36 36
                     f.write('exectue is successfully starts at ' + str(r_time) + '\n')
37 37
                     f.close()
38
-                    cmd_list = ['docker run --rm --name extract_1 --gpus '"device=0"' -v /dev:/dev -v /home/mc365/sources:/sources -v /hdd:/hdd ellishuntingmoon/mailsys:0.5 python3 /sources/extract_01.py ' + id + '/ >> /dev/null', 'echo ext_1 start']
38
+                    cmd_list = ['docker run --rm --name extract_2 --gpus '"device=1"' -v /dev:/dev -v /home/mc365/sources:/sources -v /hdd:/hdd ellishuntingmoon/mailsys:0.5 python3 /sources/extract_02.py ' + id + '/ >> /dev/null', 'echo ext_1 start']
39 39
                     for index in range(0, len(cmd_list)):
40 40
                         thread_list.append(threading.Thread(target=cmd_run,args=(index,cmd_list)))
41 41
                         thread_list[index].start()         
42
-                    os.system('docker run --rm --name extract_2 --gpus '"device=1"' -v /dev:/dev -v /home/mc365/sources:/sources -v /hdd:/hdd ellishuntingmoon/mailsys:0.5 python3 /sources/extract_02.py ' + id + '/ >> /dev/null')       
42
+                    os.system('docker run --rm --name extract_1 --gpus '"device=0"' -v /dev:/dev -v /home/mc365/sources:/sources -v /hdd:/hdd ellishuntingmoon/mailsys:0.5 python3 /sources/extract_01.py ' + id + '/ >> /dev/null')       
43 43
                     f = open(file, 'at')
44 44
                     r_time = time.ctime(time.time())
45 45
                     f.write('exectue is successfully ends at ' + str(r_time) + '\n')

Загрузка…
Отмена
Сохранить