GPU
Deep Learning小舟
深度学习算法工程师
展开
-
查询计算机上是否有可用的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 · 7842 阅读 · 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 · 5386 阅读 · 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 · 8884 阅读 · 1 评论