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!
@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.
Komentar yang paling membantu
Selain itu, di masa mendatang, StackOverflow adalah tempat yang tepat untuk mengajukan pertanyaan ini.