【转】查看tensorflow是GPU/CPU版本
查看tensorflow 是GPU版本 还是CPU版本
import os
from tensorflow.python.client import device_lib
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"
if __name__ == "__main__":
print(device_lib.list_local_devices())
...
转载
2018-10-17 09:48:06 ·
5933 阅读 ·
0 评论