Tensorflow: Tensorflow tidak menetapkan tugas apa pun ke GPU dalam contoh cifar10

Dibuat pada 20 Nov 2015 · 3Komentar · Sumber: tensorflow/tensorflow

Saya menjalankan skrip cifar10_multi_gpu_train.py dengan logging penempatan perangkat dihidupkan, dan saya melihat bahwa semua operasi ditempatkan pada CPU. Ketika saya menjalankannya, itu menghasilkan:

Filling queue with 20000 CIFAR images before starting to train. This will take a few minutes.
I tensorflow/core/common_runtime/local_device.cc:25] Local device intra op parallelism threads: 8
I tensorflow/core/common_runtime/direct_session.cc:45] Direct session inter op parallelism threads: 8
Device mapping: no known devices.
I tensorflow/core/common_runtime/direct_session.cc:111] Device mapping:

softmax_linear/biases/ExponentialMovingAverage: /job:localhost/replica:0/task:0/cpu:0
I tensorflow/core/common_runtime/simple_placer.cc:289] softmax_linear/biases/ExponentialMovingAverage: /job:localhost/replica:0/task:0/cpu:0
.....
.....

Saya akan membayangkan bahwa baris Device mapping: no known devices. seharusnya mencantumkan perangkat yang terkait dengan Sesi saat ini, tetapi mengapa tidak memiliki perangkat apa pun? Script memanggil
tf.device('/gpu:0') .

GPU saya Nvidia GeForce GTX 970.

Sumber

jricheimer

Komentar yang paling membantu

Memecahkan ini.

Saya pasti tidak sengaja membuat paket python tanpa flag "--config=cuda". Ketika saya membangun kembali dengan flag ini, lalu pip mencopot tensorflow, lalu menginstal ulang dengan paket yang baru dibuat, itu bekerja dengan sempurna.

jricheimer pada 23 Nov 2015

👍2 😕1

Semua 3 komentar

Memecahkan ini.

jricheimer pada 23 Nov 2015

👍2 😕1

Masalah yang sama terjadi ketika saya menjalankan skrip cifar10_multi_gpu_train.py dengan logging penempatan perangkat dihidupkan. Saya tidak tahu apa maksud Anda ketika Anda mengatakan "Saya pasti tidak sengaja membuat paket python tanpa tanda "--config=cuda", dapatkah Anda memberi tahu paket python mana yang Anda buat kembali?