【问题标题】:Tensorflow 2 can't find gpuTensorflow 2找不到gpu
【发布时间】:2020-03-12 15:18:37
【问题描述】:

操作系统:Ubuntu 18.04

CPU : Intel i7-8core

GPU:英伟达 GTX 1060

内存:16 GB

我已经使用以下命令在我的 anaconda 虚拟环境中安装了 tensorflow 2:

pip unistall tensorflow-gpu==2.0

安装没有错误,但是当我使用时:

tf.test.is_gpu_available()

我收到以下消息:

I tensorflow/core/platform/cpu_feature_guard.cc:142] 您的 CPU 支持未编译此 TensorFlow 二进制文件以使用的指令:AVX2 FMA
2019-11-16 20:36:36.130788: I tensorflow/core/platform/profile_utils/cpu_utils.cc:94] CPU 频率:2808000000 Hz
2019-11-16 20:36:36.131397:我 tensorflow/compiler/xla/service/service.cc:168] XLA 服务 0x55bd440b6050 在平台主机上执行计算。设备:
2019-11-16 20:36:36.131414:I tensorflow/compiler/xla/service/service.cc:175] StreamExecutor 设备(0):主机,默认版本
2019-11-16 20:36:36.133350: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] 成功打开动态库 libcuda.so.1
2019-11-16 20:36:36.134418:E tensorflow/stream_executor/cuda/cuda_driver.cc:318] 调用 cuInit 失败:CUDA_ERROR_UNKNOWN:未知错误
2019-11-16 20:36:36.134439:我 tensorflow/stream_executor/cuda/cuda_diagnostics.cc:169] 检索主机的 CUDA 诊断信息:quarmaine
2019-11-16 20:36:36.134445:我 tensorflow/stream_executor/cuda/cuda_diagnostics.cc:176] 主机名:quarmaine
2019-11-16 20:36:36.134490:我 tensorflow/stream_executor/cuda/cuda_diagnostics.cc:200] libcuda 报告的版本是:430.50.0
2019-11-16 20:36:36.134511:我 tensorflow/stream_executor/cuda/cuda_diagnostics.cc:204] 内核报告的版本是:430.50.0
2019-11-16 20:36:36.134518: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:310] 内核版本似乎匹配 DSO: 430.50.0
错误的

这表明没有可用的 GPU。

【问题讨论】:

    标签: installation tensorflow2.0


    【解决方案1】:

    首先尝试重新启动您的系统,因为可能有一些待处理的更新。 其次试试这个:-

    import os
    os.environ['CUDA_VISIBLE_DEVICES'] = "0"
    

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2021-03-21
      • 1970-01-01
      • 1970-01-01
      • 2022-10-15
      • 2018-08-30
      相关资源
      最近更新 更多