Collapse file tree

Files

 master

Search this repository

/

testGPU.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

10 lines (8 loc) · 372 Bytes

/

testGPU.py

Top

File metadata and controls

• Code
• Blame

10 lines (8 loc) · 372 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

def get_available_gpus(): """ code from http://stackoverflow.com/questions/38559755/how-to-get-current-available-gpus-in-tensorflow """ from tensorflow.python.client import device_lib as _device_lib local_device_protos = _device_lib.list_local_devices() return [x.name for x in local_device_protos if x.device_type == 'GPU'] get_available_gpus()

1

2

3

4

5

6

7

8

9