質問編集履歴
3
hgkjhhjk
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -10,7 +10,7 @@
|
|
|
10
10
|
|
|
11
11
|
|
|
12
12
|
|
|
13
|
-
GPU id を0,2,4に設定された！
|
|
13
|
+
⇓ GPU id を0,2,4に設定された！
|
|
14
14
|
|
|
15
15
|
device = torch.device(f"cuda:{min(gpus)}" if len(gpus) > 0 else 'cpu')
|
|
16
16
|
|
|
@@ -20,7 +20,7 @@
|
|
|
20
20
|
|
|
21
21
|
|
|
22
22
|
|
|
23
|
-
GPU を1,3,5に指定された！
|
|
23
|
+
⇓ GPU を1,3,5に指定された！
|
|
24
24
|
|
|
25
25
|
model = DataParallel(model, device_ids=[1,3,5])
|
|
26
26
|
|
2
修正
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -6,17 +6,21 @@
|
|
|
6
6
|
|
|
7
7
|
|
|
8
8
|
|
|
9
|
-
gpus = (0,2,4) #使用する
|
|
9
|
+
gpus = (0,2,4) #使用するGPU番号のtuple(複数選択可)
|
|
10
10
|
|
|
11
11
|
|
|
12
12
|
|
|
13
|
-
|
|
13
|
+
GPU id を0,2,4に設定された！
|
|
14
14
|
|
|
15
15
|
device = torch.device(f"cuda:{min(gpus)}" if len(gpus) > 0 else 'cpu')
|
|
16
16
|
|
|
17
17
|
|
|
18
18
|
|
|
19
|
+
----------------------------------------------
|
|
20
|
+
|
|
21
|
+
|
|
22
|
+
|
|
19
|
-
|
|
23
|
+
GPU を1,3,5に指定された！
|
|
20
24
|
|
|
21
25
|
model = DataParallel(model, device_ids=[1,3,5])
|
|
22
26
|
|
1
asdあｓｄ
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -8,8 +8,18 @@
|
|
|
8
8
|
|
|
9
9
|
gpus = (0,2,4) #使用するgpu番号のtuple(複数選択可)
|
|
10
10
|
|
|
11
|
-
device = torch.device(f"cuda:{min(gpus)}" if len(gpus) > 0 else 'cpu') # GPU id を0,2,4に設定された！
|
|
12
11
|
|
|
12
|
+
|
|
13
|
+
# GPU id を0,2,4に設定された！
|
|
14
|
+
|
|
15
|
+
device = torch.device(f"cuda:{min(gpus)}" if len(gpus) > 0 else 'cpu')
|
|
16
|
+
|
|
17
|
+
|
|
18
|
+
|
|
19
|
+
# GPU を1,3,5に指定された！
|
|
20
|
+
|
|
13
|
-
model = DataParallel(model, device_ids=[1,3,5])
|
|
21
|
+
model = DataParallel(model, device_ids=[1,3,5])
|
|
22
|
+
|
|
23
|
+
|
|
14
24
|
|
|
15
25
|
model.to(device)
|