回答編集履歴
3
d
test
CHANGED
|
@@ -20,7 +20,7 @@
|
|
|
20
20
|
|
|
21
21
|
|
|
22
22
|
|
|
23
|
-
GPU_ID=1 の GPU を使う方法
|
|
23
|
+
GPU_ID=1 の GPU を使う方法 `device = torch.device("cuda", index=1)`
|
|
24
24
|
|
|
25
25
|
|
|
26
26
|
|
2
d
test
CHANGED
|
@@ -13,3 +13,15 @@
|
|
|
13
13
|
BATCH_SIZE = 256
|
|
14
14
|
|
|
15
15
|
```
|
|
16
|
+
|
|
17
|
+
|
|
18
|
+
|
|
19
|
+
## 追記
|
|
20
|
+
|
|
21
|
+
|
|
22
|
+
|
|
23
|
+
GPU_ID=1 の GPU を使う方法
|
|
24
|
+
|
|
25
|
+
|
|
26
|
+
|
|
27
|
+
[Pytorch - 計算を行うデバイスを指定する方法について](https://pystyle.info/pytorch-how-to-specify-the-device-for-calculation/)
|
1
f
test
CHANGED
|
@@ -1,4 +1,8 @@
|
|
|
1
|
-
CUDA error: out of memory なので、単純にメモリが足りない
|
|
1
|
+
CUDA error: out of memory なので、単純に GPU のメモリが足りない
|
|
2
|
+
|
|
3
|
+
無料の Goolge Colab で割り当てられる Tesla K80 とかでも 24GB のメモリがあったと思いますが、市販のゲーミング用の GPU だと多くてもメモリ量が 12GB とかなので、業務用の GPU に比べるとメモリ量が少ない
|
|
4
|
+
|
|
5
|
+
|
|
2
6
|
|
|
3
7
|
以下のバッチサイズをエラーがでなくなるまで小さい値にしてください
|
|
4
8
|
|