1. 程式人生 > >求助Tensorflow下跑mnist手寫體資料集遇到Cuda compute capability問題

求助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中如何解決這一問題呢?謝謝!