Tensorflow 未在 PyCharm 中的 GPU 上运行

Question

我安装了Tensorflow GPU，我可以在Ubuntu 18.04的终端中看到它。

>>> from tensorflow.python.client import device_lib
>>> print(device_lib.list_local_devices())

结果是：

2018-08-21 16:34:49.867482: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1484] Adding visible gpu devices: 0
2018-08-21 16:34:49.867602: I tensorflow/core/common_runtime/gpu/gpu_device.cc:965] Device interconnect StreamExecutor with strength 1 edge matrix:
2018-08-21 16:34:49.867627: I tensorflow/core/common_runtime/gpu/gpu_device.cc:971]      0 
2018-08-21 16:34:49.867650: I tensorflow/core/common_runtime/gpu/gpu_device.cc:984] 0:   N 
2018-08-21 16:34:49.867979: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1097] Created TensorFlow device (/device:GPU:0 with 6863 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1070 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1)
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 10929943929948782169
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 7196478669
locality {
bus_id: 1
links {
}
}
incarnation: 1784950594023049308
physical_device_desc: "device: 0, name: GeForce GTX 1070 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1"
]

但是当我在 PyCharm 中运行它时，我得到

/usr/lib/python3.6/importlib/_bootstrap.py:219: RuntimeWarning: numpy.dtype size changed, may indicate binary incompatibility. Expected 96, got 88
return f(*args, **kwds)
/usr/lib/python3.6/importlib/_bootstrap.py:219: RuntimeWarning: numpy.dtype size changed, may indicate binary incompatibility. Expected 96, got 88
return f(*args, **kwds)
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 12856835725849494278
]
2018-08-21 16:38:54.238307: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
Process finished with exit code 0

Answer 1

我已经修复了它。我在 PyCharm 中更改了项目解释器，现在它对我来说是工作。