求助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中如何解决这一问题呢?谢谢!
问题所在清楚的,就是不知道有没有什么方法可以解决,让GPU跑起来?
参考技术A 遇到了同样的问题,看来只能先玩CPU版的了,等以后换电脑再玩GPU的吧。附Cuda compute capability表
https://developer.nvidia.com/cuda-gpus
compute capability代表着运算层级,看来以后要上船了。 参考技术B 没必要一定GPU吧 直接安装一个CPU版本的好了追问
GPU快呀,而且会快很多。
以上是关于求助Tensorflow下跑mnist手写体数据集遇到Cuda compute capability问题的主要内容,如果未能解决你的问题,请参考以下文章