Collapse file tree

Files

 master

Search this repository

/

testGPU2.py

Copy path

BlameMore file actions

BlameMore file actions

Latest commit

Zhen Li

检测tesnorfow的GPU是否可用

Sep 18, 2017

c1a92d8 · Sep 18, 2017

History

History

Open commit details

9 lines (7 loc) · 209 Bytes

/

testGPU2.py

Top

File metadata and controls

• Code
• Blame

9 lines (7 loc) · 209 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

import tensorflow as tf with tf.Session() as sess: with tf.device("/gpu:1"): ma1 = tf.constant([[3., 3.]]) ma2 = tf.constant([[2.],[2.]]) res = tf.matmul(ma1, ma2) print(sess.run(res))

1

2

3

4

5

6

7

8

9