我正在使用Which TensorFlow and CUDA version combinations are compatible?中提到的TensorFlow-GPU 1.12.0版本,CUDA版本9,CUDNN版本7.0.5,Bazel版本0.15和python版本3.5.2,以实现兼容性。该计算机具有Nvidia驱动程序384.130。
但是通过运行以下命令Tensorflow无法检测到我机器的GPU
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
输出如下:
[
名称:“ / device:CPU:0”device_type:“ CPU”memory_limit:268435456
locality {}
化身:13408836213255819255,名称:“ / device:XLA_CPU:0”device_type:“ XLA_CPU”memory_limit:17179869184
locality {}
化身:17981738971591465658physical_device_desc:“设备:XLA_CPU设备”]
我尝试了以下链接中的解决方案
此解决方案适用于我的情况。