您的位置:首页 > 编程语言 > Python开发

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

2017-03-11 17:32 525 查看
在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中如何解决这一问题呢?谢谢!
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息