2017-11-17 110 views
0

ここで実際のnoob - GPU上で動作するtensorflowを使用してCNNを構築しようとしているが、それはCPU上で動作し、コードは正常に動作します)。私は、GPU上でコードを実行するとGPUでTensorflowを使用するとカーネルが中断され、エラーが発生する: "... TensorFlowバイナリはコンパイルされていない:AVX AVX2"

device_name = "GPU:0" 
with tf.device(device_name): 
    # Code here 

事が始めるが、数秒後に、私は(これは単なる予告ではない - それは実際に実行し続けることから、コードを停止します)、次のタイトルのエラーが出ます:

An error ocurred while starting the kernel 

と次の詳細(それは長いと繰り返しているだけなので最初の数行を含む)

2017 11:37:20.918191: I C:\tf_jenkins\home\workspace\rel‑win\M\windows‑gpu\PY\35\tensorflow\core\platform\cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2 
2017 11:37:21.959053: I C:\tf_jenkins\home\workspace\rel‑win\M\windows‑gpu\PY\35\tensorflow\core\common_runtime\gpu\gpu_device.cc:1030] Found device 0 with properties: 
name: GeForce GTX 1050 Ti major: 6 minor: 1 memoryClockRate(GHz): 1.62 
pciBusID: 0000:01:00.0 
totalMemory: 4.00GiB freeMemory: 3.29GiB 
2017 11:37:21.959079: I C:\tf_jenkins\home\workspace\rel‑win\M\windows‑gpu\PY\35\tensorflow\core\common_runtime\gpu\gpu_device.cc:1120] Creating TensorFlow device (/device:GPU:0) ‑> (device: 0, name: GeForce GTX 1050 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1) 
2017 11:37:22.119554: I C:\tf_jenkins\home\workspace\rel‑win\M\windows‑gpu\PY\35\tensorflow\core\common_runtime\direct_session.cc:299] Device mapping: 
/job:localhost/replica:0/task:0/device:GPU:0 ‑> device: 0, name: GeForce GTX 1050 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1 

私はそれがデバイスを見つけ嬉しく思います - 重要なビットがあると思われます。

Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2 

私は

import os 
os.environ['TF_CPP_MIN_LOG_LEVEL']='2' 
import tensorflow as tf 

を使用して警告を無視しようとしたが、それは何も変わりません。

私はまだ基​​礎を学んでいるので、すべての指針は非常に高く評価されます。ありがとうございました。

答えて

0

ここでは2つの異なることが起こっていると思います。デバイス名を指定しなかったのですか? TensorFlowは、アクセス可能なすべてのGPUを自動的に使用します。

AVXの問題は、おそらくpipを使用してTensorFlowをインストールしたためですが、bazelを使用してTensorFlowをコンパイルすると、その警告は消えてしまうでしょう。

「繰り返し」スタックトレースを見ることなく、何が起こっているのかを正確に知ることが難しくなります。

関連する問題