求助Tensorflow下跑mnist手写体数据集遇到Cuda compute capability问题

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了求助Tensorflow下跑mnist手写体数据集遇到Cuda compute capability问题相关的知识,希望对你有一定的参考价值。

在Python下装了tensorflow-gpu,其中cuda为cuda_8.0.61_windows,cudnn为cudnn-8.0-windows7-x64-v5.1,安装没有问题,可以正常跑起来,但是在跑mnist手写体数据集时遇到以下问题:
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\core\common_runtime\gpu\gpu_device.cc:948] Ignoring visible gpu device (device: 0, name: GeForce GT 630M, pci bus id: 0000:01:00.0) with Cuda compute capability 2.1. The minimum required Cuda capability is 3.0.
究其原因是显卡不支持Cuda compute capability 3.0,看到网上caffe中此类问题的解决方法是在Makefile.config中注释掉USE_CUDNN。请问,哪位大神知道Tensorflow中如何解决这一问题呢?谢谢!

因为你的gt630m性能没有达到最低的3.0要求。所以报错。建议你用cpu。你的在个显卡,加速效果可以忽略不计。另外用它加速计算,还容易导致笔记本散热不畅损坏GPU。追问

问题所在清楚的,就是不知道有没有什么方法可以解决,让GPU跑起来?

参考技术A 遇到了同样的问题,看来只能先玩CPU版的了,等以后换电脑再玩GPU的吧。
附Cuda compute capability表

https://developer.nvidia.com/cuda-gpus
compute capability代表着运算层级,看来以后要上船了。
参考技术B 没必要一定GPU吧 直接安装一个CPU版本的好了追问

GPU快呀,而且会快很多。

以上是关于求助Tensorflow下跑mnist手写体数据集遇到Cuda compute capability问题的主要内容,如果未能解决你的问题,请参考以下文章

Mnist手写数字识别 Tensorflow

Tensorflow可视化MNIST手写数字训练

TensorFlow 手写体数字识别

Tensorflow2.0的手写数字识别系统(Mnist数据集)

使用tensorflow操作MNIST数据

tensorflow 基础学习五:MNIST手写数字识别