Tensorflow: Bisakah saya mengekspor model terlatih gpu ke mesin khusus cpu untuk penyajian/evaluasi?

Dibuat pada 16 Jul 2016  ·  3Komentar  ·  Sumber: tensorflow/tensorflow

Saya melatih model pada mesin GPU.
Ketika saya memuat model seperti itu pada mesin khusus cpu, saya mendapatkan kesalahan di bawah ini.

Terima kasih atas saran apa pun!

awaiting tensorflower

Komentar yang paling membantu

Selain itu, di masa mendatang, StackOverflow adalah tempat yang tepat untuk mengajukan pertanyaan ini.

Semua 3 komentar

@vrv Bisakah Anda melihat ini? Apakah kita memiliki dukungan seperti itu sekarang? Terima kasih.

Yup, Anda dapat meneruskan tf.ConfigProto(allow_soft_placement=True) ke tf.Session jika Anda benar-benar ingin mengabaikan arahan penempatan perangkat dalam grafik. Saya bukan penggemar berat menggunakan opsi ini (saya lebih suka grafik ditulis ulang untuk menghapus bidang perangkat secara eksplisit), tetapi itu akan berhasil.

Selain itu, di masa mendatang, StackOverflow adalah tempat yang tepat untuk mengajukan pertanyaan ini.

Apakah halaman ini membantu?
0 / 5 - 0 peringkat