質問編集履歴
1
ああああ
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -1,6 +1,26 @@
|
|
|
1
|
-
tensorflow v1でGPUを使って計算をさせていて、と
|
|
1
|
+
tensorflow v1でGPUを使って計算をさせていて、とコードAを入れて走らせたところsegmentation fault とでてとまってしまいました。
|
|
2
2
|
|
|
3
3
|
同じコードをCPUで走らせたらエラーなく動きました。
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
|
|
7
|
+
codeA
|
|
8
|
+
|
|
9
|
+
V = tf.Variable([batch_size, 7], dtype=tf.float32)
|
|
10
|
+
|
|
11
|
+
for b in range(B):
|
|
12
|
+
|
|
13
|
+
xxxxxxxx
|
|
14
|
+
|
|
15
|
+
for i, num in enumerate(nums):
|
|
16
|
+
|
|
17
|
+
A = tf.where(tf.equal(A[b], num), tf.broadcast_to(V[b,i], [224, 224]), tf.broadcast_to(0., [224, 224]))
|
|
18
|
+
|
|
19
|
+
yyyyyyy
|
|
20
|
+
|
|
21
|
+
zzzzzzz
|
|
22
|
+
|
|
23
|
+
|
|
4
24
|
|
|
5
25
|
|
|
6
26
|
|
|
@@ -58,7 +78,9 @@
|
|
|
58
78
|
|
|
59
79
|
```
|
|
60
80
|
|
|
61
|
-
|
|
81
|
+
コードAには tf.broadcast_toを使っています。でその対象は一つはtf.Variable(float32), もう一つは 0. です。
|
|
82
|
+
|
|
83
|
+
やっぱりそこが原因なのでしょうか？
|
|
62
84
|
|
|
63
85
|
CPUでは動いたのに。。。
|
|
64
86
|
|