|
1 2 3 4 |
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信息：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
>>> 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 |