测试tensorflow-gpu是否安装成功

>>> import tensorflow as tf>>> tf.test.is_gpu_available()结果如下：(tf14) C:\Users\Administrator>pythonPython 3.6.13 |Anaconda, Inc.| (default, Mar 16 2021, 11:37:27) [MSC v.1916 64 bi

Jennifer233

6131人浏览 · 2021-08-04 16:46:55

Jennifer233 · 2021-08-04 16:46:55 发布

>>> import tensorflow as tf
>>> tf.test.is_gpu_available()

结果如下：

(tf14) C:\Users\Administrator>python
Python 3.6.13 |Anaconda, Inc.| (default, Mar 16 2021, 11:37:27) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> tf.test.is_gpu_available()
2021-08-04 16:44:11.325631: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
2021-08-04 16:44:11.457415: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1433] Found device 0 with properties:
name: GeForce GTX 1660 SUPER major: 7 minor: 5 memoryClockRate(GHz): 1.8
pciBusID: 0000:01:00.0
totalMemory: 6.00GiB freeMemory: 5.05GiB
2021-08-04 16:44:11.457619: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1512] Adding visible gpu devices: 0
2021-08-04 16:44:11.841519: I tensorflow/core/common_runtime/gpu/gpu_device.cc:984] Device interconnect StreamExecutor with strength 1 edge matrix:
2021-08-04 16:44:11.841672: I tensorflow/core/common_runtime/gpu/gpu_device.cc:990]      0
2021-08-04 16:44:11.842694: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1003] 0:   N
2021-08-04 16:44:11.843244: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1115] Created TensorFlow device (/device:GPU:0 with 4773 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1660 SUPER, pci bus id: 0000:01:00.0, compute capability: 7.5)
True
>>>