[mcj]测试TensorFlow-GPU是否安装成功-马春杰杰

Python 3.6.2 |Continuum Analytics, Inc.| (default, Jul 20 2017, 13:51:32) 
[GCC 4.4.7 20120313 (Red Hat 4.4.7-1)] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf

Python 3.6.2 |Continuum Analytics, Inc.| (default, Jul 20 2017, 13:51:32)

[GCC 4.4.7 20120313 (Red Hat 4.4.7-1)] on linux

Type "help", "copyright", "credits" or "license" for more information.

>>> import tensorflow as tf

然后测试TensorFlow-GPU是否可用：

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

如果可用的话，会输出GPU信息：

>>> sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
2020-01-09 04:07:09.205315: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA
2020-01-09 04:07:09.649952: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1405] Found device 0 with properties: 
name: GeForce RTX 2080 Ti major: 7 minor: 5 memoryClockRate(GHz): 1.545
pciBusID: 0000:02:00.0
totalMemory: 10.73GiB freeMemory: 10.57GiB
2020-01-09 04:07:09.650001: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1484] Adding visible gpu devices: 0
2020-01-09 04:07:10.330487: I tensorflow/core/common_runtime/gpu/gpu_device.cc:965] Device interconnect StreamExecutor with strength 1 edge matrix:
2020-01-09 04:07:10.330525: I tensorflow/core/common_runtime/gpu/gpu_device.cc:971]      0 
2020-01-09 04:07:10.330535: I tensorflow/core/common_runtime/gpu/gpu_device.cc:984] 0:   N 
2020-01-09 04:07:10.331278: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1097] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 10211 MB memory) -> physical GPU (device: 0, name: GeForce RTX 2080 Ti, pci bus id: 0000:02:00.0, compute capability: 7.5)
Device mapping:
/job:localhost/replica:0/task:0/device:GPU:0 -> device: 0, name: GeForce RTX 2080 Ti, pci bus id: 0000:02:00.0, compute capability: 7.5
2020-01-09 04:07:10.457012: I tensorflow/core/common_runtime/direct_session.cc:288] Device mapping:
/job:localhost/replica:0/task:0/device:GPU:0 -> device: 0, name: GeForce RTX 2080 Ti, pci bus id: 0000:02:00.0, compute capability: 7.5

>>> sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

2020-01-09 04:07:09.205315: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA

2020-01-09 04:07:09.649952: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1405] Found device 0 with properties:

name: GeForce RTX 2080 Ti major: 7 minor: 5 memoryClockRate(GHz): 1.545

pciBusID: 0000:02:00.0

totalMemory: 10.73GiB freeMemory: 10.57GiB

2020-01-09 04:07:09.650001: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1484] Adding visible gpu devices: 0

2020-01-09 04:07:10.330487: I tensorflow/core/common_runtime/gpu/gpu_device.cc:965] Device interconnect StreamExecutor with strength 1 edge matrix:

2020-01-09 04:07:10.330525: I tensorflow/core/common_runtime/gpu/gpu_device.cc:971] 0

2020-01-09 04:07:10.330535: I tensorflow/core/common_runtime/gpu/gpu_device.cc:984] 0: N

2020-01-09 04:07:10.331278: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1097] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 10211 MB memory) -> physical GPU (device: 0, name: GeForce RTX 2080 Ti, pci bus id: 0000:02:00.0, compute capability: 7.5)

Device mapping: