Tensorflow'un GPU sürümünü Ubuntu 14.04'e yükledim.CPU'da Tensorflow nasıl çalıştırılır
Tensorflow'un mevcut GPU'lara erişebildiği bir GPU sunucustayım.
CPU'larda tensorflow çalıştırmak istiyorum.
Normalde, env CUDA_VISIBLE_DEVICES=0 numaralı telefonu, GPU no. 0.
Bunun yerine CPU'ları arasında nasıl seçim yapabilirim?
Ben Sen uygulayabilirsiniz with tf.device("/cpu:0"):
ile kodumu yeniden yazabilecek intersted değilim device_count parametre tf.Session başına:
config = tf.ConfigProto(
device_count = {'GPU': 0}
)
sess = tf.Session(config=config)
Ayrıca bkz protobuf yapılandırma dosyası:
yapabilirsiniz ayrıca ortam değişkenini
olarak ayarlayın. Kaynak kodu değiştirmek zorunda kalmadan,.
^^ Bu doğru cevaptır. –
Bu da benim için çalıştı. – VAndrei
Birisi, eğitim aşamasından sonra GPU'lar üzerinde çalıştığı kadar yüksek performanslı CPU'ları çalıştırarak, yani sadece eğitim ifadesinin GPU'ya ihtiyaç duyduğunu söyledi. Bunun doğru olup olmadığını biliyor musun? Teşekkürler! – Crashalot
Yukarıdaki cevaplar çalışmazsa, birini deneyin:
os.environ['CUDA_VISIBLE_DEVICES'] = ''
veya
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
Teşekkürler, bu diğer seçeneklerden çok daha iyi. – user1098761
Birisi eğitim faz GPU'larda bunları çalıştıran kadar etkili olduğunu sonra CPU üzerindeki nöral ağları çalışan sözü - - Yani, sadece eğitim ifadesi gerçekten GPU'ya ihtiyaç duyar. Bunun doğru olup olmadığını biliyor musun? Teşekkürler! – Crashalot
bu benim için çalışıyor. Çok basit – Shaowu
Bu benim için çalışmıyor (tf1.1). FabrizioM çözümü yapar. – user1735003