1+ {
2+ "cells" : [
3+ {
4+ "cell_type" : " code" ,
5+ "execution_count" : 10 ,
6+ "metadata" : {
7+ "collapsed" : true
8+ },
9+ "outputs" : [],
10+ "source" : [
11+ " import numpy as np\n " ,
12+ " import tensorflow as tf\n " ,
13+ " import time"
14+ ]
15+ },
16+ {
17+ "cell_type" : " markdown" ,
18+ "source" : [
19+ " ## Detecting GPUs\n "
20+ ],
21+ "metadata" : {
22+ "collapsed" : false ,
23+ "pycharm" : {
24+ "name" : " #%% md\n "
25+ }
26+ }
27+ },
28+ {
29+ "cell_type" : " code" ,
30+ "execution_count" : 11 ,
31+ "outputs" : [
32+ {
33+ "name" : " stdout" ,
34+ "output_type" : " stream" ,
35+ "text" : [
36+ " [name: \" /device:CPU:0\"\n " ,
37+ " device_type: \" CPU\"\n " ,
38+ " memory_limit: 268435456\n " ,
39+ " locality {\n " ,
40+ " }\n " ,
41+ " incarnation: 11702401161608791080\n " ,
42+ " , name: \" /device:XLA_CPU:0\"\n " ,
43+ " device_type: \" XLA_CPU\"\n " ,
44+ " memory_limit: 17179869184\n " ,
45+ " locality {\n " ,
46+ " }\n " ,
47+ " incarnation: 2978008688873522925\n " ,
48+ " physical_device_desc: \" device: XLA_CPU device\"\n " ,
49+ " ]\n "
50+ ]
51+ }
52+ ],
53+ "source" : [
54+ " from tensorflow.python.client import device_lib\n " ,
55+ " print(device_lib.list_local_devices())\n "
56+ ],
57+ "metadata" : {
58+ "collapsed" : false ,
59+ "pycharm" : {
60+ "name" : " #%%\n "
61+ }
62+ }
63+ },
64+ {
65+ "cell_type" : " markdown" ,
66+ "source" : [
67+ " ## performance Comparison"
68+ ],
69+ "metadata" : {
70+ "collapsed" : false ,
71+ "pycharm" : {
72+ "name" : " #%% md\n "
73+ }
74+ }
75+ },
76+ {
77+ "cell_type" : " code" ,
78+ "execution_count" : 12 ,
79+ "outputs" : [],
80+ "source" : [
81+ " ITER = 500\n " ,
82+ " X = np.random.randn(1000,1000)"
83+ ],
84+ "metadata" : {
85+ "collapsed" : false ,
86+ "pycharm" : {
87+ "name" : " #%%\n "
88+ }
89+ }
90+ },
91+ {
92+ "cell_type" : " markdown" ,
93+ "source" : [
94+ " ### Numpy performance\n "
95+ ],
96+ "metadata" : {
97+ "collapsed" : false ,
98+ "pycharm" : {
99+ "name" : " #%% md\n "
100+ }
101+ }
102+ },
103+ {
104+ "cell_type" : " code" ,
105+ "execution_count" : 13 ,
106+ "outputs" : [
107+ {
108+ "name" : " stdout" ,
109+ "output_type" : " stream" ,
110+ "text" : [
111+ " Time taken by numpy8.148992\n "
112+ ]
113+ }
114+ ],
115+ "source" : [
116+ " start = time.time()\n " ,
117+ " for i in range(ITER):\n " ,
118+ " out = np.matmul(X,X)\n " ,
119+ " \n " ,
120+ " end = time.time()\n " ,
121+ " \n " ,
122+ " print(\" Time taken by numpy{:.6f}\" .format(end-start))"
123+ ],
124+ "metadata" : {
125+ "collapsed" : false ,
126+ "pycharm" : {
127+ "name" : " #%%\n "
128+ }
129+ }
130+ },
131+ {
132+ "cell_type" : " markdown" ,
133+ "source" : [
134+ " ### Tensorflow performance"
135+ ],
136+ "metadata" : {
137+ "collapsed" : false ,
138+ "pycharm" : {
139+ "name" : " #%% md\n "
140+ }
141+ }
142+ },
143+ {
144+ "cell_type" : " code" ,
145+ "execution_count" : 14 ,
146+ "outputs" : [],
147+ "source" : [
148+ " import tensorflow.compat.v1 as tf\n " ,
149+ " tf.disable_v2_behavior()"
150+ ],
151+ "metadata" : {
152+ "collapsed" : false ,
153+ "pycharm" : {
154+ "name" : " #%%\n "
155+ }
156+ }
157+ },
158+ {
159+ "cell_type" : " code" ,
160+ "execution_count" : 15 ,
161+ "outputs" : [],
162+ "source" : [
163+ " x = tf.placeholder( shape=(1000,1000),dtype=tf.float32)\n " ,
164+ " output = tf.matmul(x,x)"
165+ ],
166+ "metadata" : {
167+ "collapsed" : false ,
168+ "pycharm" : {
169+ "name" : " #%%\n "
170+ }
171+ }
172+ },
173+ {
174+ "cell_type" : " code" ,
175+ "execution_count" : 16 ,
176+ "outputs" : [
177+ {
178+ "name" : " stdout" ,
179+ "output_type" : " stream" ,
180+ "text" : [
181+ " Time taken by Tensorflow --> 4.072855\n "
182+ ]
183+ }
184+ ],
185+ "source" : [
186+ " sess = tf.Session()\n " ,
187+ " start = time.time()\n " ,
188+ " for i in range(ITER):\n " ,
189+ " sess.run(output, feed_dict={\n " ,
190+ " x: X\n " ,
191+ " })\n " ,
192+ " end = time.time()\n " ,
193+ " \n " ,
194+ " print(\" Time taken by Tensorflow --> {:.6f}\" .format(end-start))"
195+ ],
196+ "metadata" : {
197+ "collapsed" : false ,
198+ "pycharm" : {
199+ "name" : " #%%\n "
200+ }
201+ }
202+ },
203+ {
204+ "cell_type" : " code" ,
205+ "execution_count" : 17 ,
206+ "outputs" : [
207+ {
208+ "name" : " stdout" ,
209+ "output_type" : " stream" ,
210+ "text" : [
211+ " Time taken by Tensorflow --> 4.189568\n "
212+ ]
213+ }
214+ ],
215+ "source" : [
216+ " with tf.device('/device:CPU:0'):\n " ,
217+ " sess = tf.Session()\n " ,
218+ " start = time.time()\n " ,
219+ " for i in range(ITER):\n " ,
220+ " sess.run(output, feed_dict={\n " ,
221+ " x: X\n " ,
222+ " })\n " ,
223+ " end = time.time()\n " ,
224+ " \n " ,
225+ " print(\" Time taken by Tensorflow --> {:.6f}\" .format(end-start))"
226+ ],
227+ "metadata" : {
228+ "collapsed" : false ,
229+ "pycharm" : {
230+ "name" : " #%%\n "
231+ }
232+ }
233+ }
234+ ],
235+ "metadata" : {
236+ "kernelspec" : {
237+ "display_name" : " Python 3" ,
238+ "language" : " python" ,
239+ "name" : " python3"
240+ },
241+ "language_info" : {
242+ "codemirror_mode" : {
243+ "name" : " ipython" ,
244+ "version" : 2
245+ },
246+ "file_extension" : " .py" ,
247+ "mimetype" : " text/x-python" ,
248+ "name" : " python" ,
249+ "nbconvert_exporter" : " python" ,
250+ "pygments_lexer" : " ipython2" ,
251+ "version" : " 2.7.6"
252+ }
253+ },
254+ "nbformat" : 4 ,
255+ "nbformat_minor" : 0
256+ }
0 commit comments