2016-12-09 1 views
0

ここでは、実行結果を、私はhttps://github.com/Russell91/TensorBoxからのコードで遊んでいます。しかし、GPUで実行に失敗していますはtensorflow.train.threading.Threadはサポートされなくなりましたか?

TensorBox$ python train.py --hypes hypes/overfeat_rezoom.json --gpu 0 --logdir output 
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcublas.so locally 
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcudnn.so locally 
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcufft.so locally 
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcuda.so.1 locally 
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcurand.so locally 
I tensorflow/core/common_runtime/gpu/gpu_device.cc:885] Found device 0 with properties: 
name: Quadro M4000 
major: 5 minor: 2 memoryClockRate (GHz) 0.7725 
pciBusID 0000:02:00.0 
Total memory: 7.93GiB 
Free memory: 7.63GiB 
I tensorflow/core/common_runtime/gpu/gpu_device.cc:906] DMA: 0 
I tensorflow/core/common_runtime/gpu/gpu_device.cc:916] 0: Y 
I tensorflow/core/common_runtime/gpu/gpu_device.cc:975] Creating TensorFlow device (/gpu:0) -> (device: 0, name: Quadro M4000, pci bus id: 0000:02:00.0) 
Traceback (most recent call last): 
    File "train.py", line 537, in <module> 
    main() 
    File "train.py", line 534, in main 
    train(H, test_images=[]) 
    File "train.py", line 457, in train 
    t = tf.train.threading.Thread(target=thread_loop, 
AttributeError: 'module' object has no attribute 'threading' 

tensorflowそれがスレッドを処理する方法を変更しましたか?それ以外の場合、私は何を間違っているのですか?

答えて

1

TL; DR:threading.Threadに変更してください。

輸入tensorflow.train.threading.Threadは、APIの正式一部ではなかったのではなく、我々はシンボルがTensorFlowモジュールを介して表示されていたかについては慎重でなかったため、アクセス可能でした。安定したリリースに向けて、Pythonの__all__のような手法を使ってモジュールの正確な内容を定義することで、このような文書化されていないインクルードを取り締まっています。

関連する問題