tensorflow
Deep Learning小舟
深度学习算法工程师
展开
-
tensorflow:OP_REQUIRES failed at save_restore_v2_ops.cc:109
OP_REQUIRES failed at save_restore_v2_ops.cc:109 : Not found: Failed to create a NewWriteableFile: ; No such process原创 2021-01-20 12:05:50 · 6870 阅读 · 7 评论 -
查询计算机上是否有可用的GPU设备
查询计算机上是否有可用的GPU设备代码结果展示代码def is_gpu_available(cuda_only=True): from tensorflow.python.client import device_lib as _device_lib if cuda_only: return any((x.device_type == 'GPU') for x in _device_lib.list_local_devices())原创 2020-11-29 15:59:31 · 7785 阅读 · 4 评论 -
获取PC 服务器 可用的GPU
获取PC 服务器 可用的GPU代码结果展示代码def get_available_gpus(): from tensorflow.python.client import device_lib as _device_lib local_device_protos = _device_lib.list_local_devices() return [x.name for x in local_device_protos if x.device_type == 'GPU']prin原创 2020-11-29 15:47:28 · 5359 阅读 · 3 评论 -
测试tensorflow_GPU是否安装成功
测试tensorflow_GPU是否安装成功代码结果测试tensorflow_GPU是否安装成功若正常应该输出2,并且在输出2之前有很长串的信息如算力(ability)、gpu的分配情况等等。代码# coding=gbk# coding=gbk# 测试tensorflow_GPU是否安装成功# 若正常应该输出2,并且在输出2之前有很长串的信息如算力(ability)、gpu的分配情况等等。import tensorflow as tffor i in range(1000): a原创 2020-11-29 15:31:20 · 8804 阅读 · 1 评论