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

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
                     r_time = time.ctime(time.time())
35
                     r_time = time.ctime(time.time())
36
                     f.write('exectue is successfully starts at ' + str(r_time) + '\n')
36
                     f.write('exectue is successfully starts at ' + str(r_time) + '\n')
37
                     f.close()
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
                     for index in range(0, len(cmd_list)):
39
                     for index in range(0, len(cmd_list)):
40
                         thread_list.append(threading.Thread(target=cmd_run,args=(index,cmd_list)))
40
                         thread_list.append(threading.Thread(target=cmd_run,args=(index,cmd_list)))
41
                         thread_list[index].start()         
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
                     f = open(file, 'at')
43
                     f = open(file, 'at')
44
                     r_time = time.ctime(time.time())
44
                     r_time = time.ctime(time.time())
45
                     f.write('exectue is successfully ends at ' + str(r_time) + '\n')
45
                     f.write('exectue is successfully ends at ' + str(r_time) + '\n')

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