首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >GPU无法通过tensorflow检测-gpu 2.0

GPU无法通过tensorflow检测-gpu 2.0
EN

Stack Overflow用户
提问于 2019-10-27 23:30:08
回答 1查看 69关注 0票数 0

GPU无法检测到tensorflow-gpu 2.0,问题出在哪里?为

代码语言:javascript
复制
import tensorflow as tf
print(tf.test.is_built_with_cuda())
print(tf.test.is_gpu_available())
from tensorflow.python.client import device_lib

device_lib.list_local_devices()

下面是使用Ubuntu 16.04的我的电脑中的输出

代码语言:javascript
复制
True
False
[name: "/device:CPU:0"
 device_type: "CPU"
 memory_limit: 268435456
 locality {
 }
 incarnation: 10185800011826802770, name: "/device:XLA_CPU:0"
 device_type: "XLA_CPU"
 memory_limit: 17179869184
 locality {
 }
 incarnation: 2908743746130792497
 physical_device_desc: "device: XLA_CPU device", name: "/device:XLA_GPU:0"
 device_type: "XLA_GPU"
 memory_limit: 17179869184
 locality {
 }
 incarnation: 13796634005997726084
 physical_device_desc: "device: XLA_GPU device", name: "/device:XLA_GPU:1"
 device_type: "XLA_GPU"
 memory_limit: 17179869184
 locality {
 }
 incarnation: 4940094218016527544
 physical_device_desc: "device: XLA_GPU device", name: "/device:XLA_GPU:2"
 device_type: "XLA_GPU"
 memory_limit: 17179869184
 locality {
 }
 incarnation: 15391692776259568109
 physical_device_desc: "device: XLA_GPU device", name: "/device:XLA_GPU:3"
 device_type: "XLA_GPU"
 memory_limit: 17179869184
 locality {
 }
 incarnation: 8368387340417357161
 physical_device_desc: "device: XLA_GPU device"]
EN

回答 1

Stack Overflow用户

发布于 2019-10-28 11:37:22

正如y.selivonchyk指出的那样,在安装CUDA10.0之后,它现在可以工作了。虽然不容易去那里(涉及卸载/重新安装nvidia驱动程序和cuda),如果有人需要帮助,请在评论中让我知道。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/58580781

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档