from tensorflow.python.client import device_lib
print(device_lib.list_local_devices()
如何查看自己使用的tensorflow是cpu还是gpu版本？
猜你喜欢
转载自blog.csdn.net/weixin_41194129/article/details/107763236
今日推荐
周排行