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