2015-11-11 6 views
18

एक g2.2xlarge EC2 उदाहरण पर TensorFlow और उसके निर्भरता स्थापित करने के बाद मैं आने शुरू हो गए पृष्ठ से एक MNIST उदाहरण भागने की कोशिश की है:न्यूनतम आवश्यक Cuda क्षमता 3.5

python tensorflow/models/image/mnist/convolutional.py 

लेकिन मैं निम्नलिखित चेतावनी मिलती है:

I tensorflow/core/common_runtime/gpu/gpu_device.cc:611] Ignoring gpu device 
(device: 0, name: GRID K520, pci bus id: 0000:00:03.0) with Cuda compute 
capability 3.0. The minimum required Cuda capability is 3.5. 

क्या यह एक कठिन आवश्यकता है? किसी भी मौके पर मैं टिप्पणी कर सकता हूं कि टेंसरफ्लो के कांटा में जांच करें? एडब्ल्यूएस में मॉडल को प्रशिक्षित करने में सक्षम होने के लिए यह बहुत अच्छा होगा।

+0

हमने गणना की क्षमता> = 3.5 के साथ केवल टेन्सरफ़्लो का परीक्षण किया है। कई लोगों ने 3.0 के लिए समर्थन का अनुरोध किया है, और यह हमारे रडार पर है: यहां गिटहब मुद्दे पर चर्चा देखें https://github.com/tensorflow/tensorflow/issues/25 – mrry

उत्तर

8

वर्तमान में गणना क्षमता के साथ केवल जीपीयू> = 3.5 आधिकारिक रूप से समर्थित हैं। हालांकि, गिटहब उपयोगकर्ता @ इनफोजंकी ने a patch की पेशकश की है जो कंप्यूट क्षमता 3.0 के साथ एक जीपीयू के साथ टेंसरफ्लो का उपयोग करना संभव बनाता है।

आधिकारिक फ़िक्स विकास में है। इस बीच, इस समर्थन को जोड़ने के लिए GitHub issue पर चर्चा देखें।

9

official installation page में एक अनुभाग है जो आपको Cuda 3 को सक्षम करने के लिए मार्गदर्शन करता है, लेकिन आपको स्रोत से Tensorflow बनाने की आवश्यकता है।

$ TF_UNOFFICIAL_SETTING=1 ./configure 

# Same as the official settings above 

WARNING: You are configuring unofficial settings in TensorFlow. Because some 
external libraries are not backward compatible, these settings are largely 
untested and unsupported. 

Please specify a list of comma-separated Cuda compute capabilities you want to 
build with. You can find the compute capability of your device at: 
https://developer.nvidia.com/cuda-gpus. 
Please note that each additional compute capability significantly increases 
your build time and binary size. [Default is: "3.5,5.2"]: 3.0 

Setting up Cuda include 
Setting up Cuda lib64 
Setting up Cuda bin 
Setting up Cuda nvvm 
Configuration finished 
+0

"[डिफ़ॉल्ट है:" 3.5,5.2 "]: 3.0 "मेरी समस्या हल! –

1

एक साधारण चाल है। आपको स्रोतों से टीएफ भी बनाना नहीं है।

फ़ाइल में tensorflow\python\_pywrap_tensorflow.pyd रेगेक्स 3\.5.*5\.2 के दो अवसर हैं। के साथ बस 3.5 दोनों को प्रतिस्थापित करें।

विंडोज 10, एनाकोंडा 4.2.13, पायथन 3.5.2, टेन्सरफ्लो 0.12, सीयूडीए 8, एनवीडिया जीटीएक्स 660 मीटर (सीयूडीए कैप 3.0) पर परीक्षण किया गया।

संबंधित मुद्दे