首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >tensorflow未拾取GPU卡- GPU

tensorflow未拾取GPU卡- GPU
EN

Stack Overflow用户
提问于 2021-01-22 05:24:52
回答 1查看 41关注 0票数 0

我安装了tensorflow-gpu版本,并尝试按照建议测试GPU设置

代码语言:javascript
复制
import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

但是,我得到了以下信息

代码语言:javascript
复制
Num GPUs Available:  0

我的机器确实有GPU卡,如下所示,为什么没有被Tensorflow选中

EN

回答 1

Stack Overflow用户

发布于 2021-01-22 05:29:36

您可能没有以下各项的正确组合:

  1. CUDA
  2. CuDNN
  3. TensorFlow

请在这里检查我的答案,以获得上述内容的正确组合:Tensorflow 2.0 can't use GPU, something wrong in cuDNN? :Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

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

https://stackoverflow.com/questions/65835954

复制
相关文章

相似问题

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