質問編集履歴
1
質問内容について意図(jaxのCPU上におけるパフォーマンスについて知りたい)と説明文が不一致していたため,修正を行いました.
test
CHANGED
@@ -1 +1 @@
|
|
1
|
-
JAXの行列計算速度がnumpyに劣ってしまう問題
|
1
|
+
CPUによる計算環境において,JAXの行列計算速度がnumpyに劣ってしまう問題
|
test
CHANGED
@@ -10,13 +10,23 @@
|
|
10
10
|
|
11
11
|
|
12
12
|
|
13
|
-
jaxの強みは高速な自動微分であると思っていたのですが,
|
13
|
+
~~jaxの強みは高速な自動微分であると思っていたのですが,
|
14
14
|
|
15
15
|
numpyよりも遅くなってしまう事例が発生しました.
|
16
16
|
|
17
17
|
皆様には,なぜjaxの計算が遅くなってしまったのかについて,
|
18
18
|
|
19
|
-
ご意見をお聞きしたいと思っております.
|
19
|
+
ご意見をお聞きしたいと思っております.~~
|
20
|
+
|
21
|
+
(追記 質問内容が意図と異なっていたため修正いたします)
|
22
|
+
|
23
|
+
google colab上で**CPUによる計算**を行わせたところ,jaxの行列計算速度がnumpyよりも遅くなってしまう事例が発生しました.
|
24
|
+
|
25
|
+
jaxの強みは高速な自動微分であると思っていたため,CPU上では計算速度が低くなってしまうことを不思議に思い,
|
26
|
+
|
27
|
+
また,jaxの知識が浅くよくわかっていないため,皆様へご意見をお聞きしたいと思います.
|
28
|
+
|
29
|
+
jaxはCPUの計算環境においては,計算速度がnumpyよりも劣ってしまうのでしょうか?
|
20
30
|
|
21
31
|
|
22
32
|
|
@@ -62,7 +72,7 @@
|
|
62
72
|
|
63
73
|
|
64
74
|
|
65
|
-
私の環境(Google Colab)で同等のコードで実行したところ,
|
75
|
+
私の環境(Google Colab/ CPU)で同等のコードで実行したところ,
|
66
76
|
|
67
77
|
Time taken by dot product op on ndarrays: 14.60 seconds
|
68
78
|
|
@@ -76,17 +86,17 @@
|
|
76
86
|
|
77
87
|
|
78
88
|
|
79
|
-
私のコードは,全てからコピペしたものであり,全く同じコードです.
|
89
|
+
~~私のコードは,全てからコピペしたものであり,全く同じコードです.
|
80
90
|
|
81
|
-
それなのに,計算速度に違いが生まれてしまうことが不思議でなりません.
|
91
|
+
それなのに,計算速度に違いが生まれてしまうことが不思議でなりません.~~
|
82
92
|
|
83
|
-
唯一出たエラーコードは
|
93
|
+
~~唯一出たエラーコードは
|
84
94
|
|
85
95
|
[1]を実行したとき:/usr/local/lib/python3.7/dist-packages/ipykernel_launcher.py:8: UserWarning: Config option `use_jedi` not recognized by `IPCompleter`.
|
86
96
|
|
87
|
-
[2]を実行したとき:WARNING:absl:No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)
|
97
|
+
[2]を実行したとき:WARNING:absl:No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)~~
|
88
98
|
|
89
|
-
jaxは計算
|
99
|
+
jaxはCPUの計算環境においては,計算速度がnumpyよりも劣ってしまうのでしょうか?
|
90
100
|
|
91
101
|
お詳しい方がいらっしゃいましたら,ご意見をいただけますと幸いです.
|
92
102
|
|