Dec 12, 2017

ติดตั้ง TensorFlow 1.4 บน Ubutu 16.04 LST ก้าวแรกสู่ Deep Learning

จากคู่มือของ TensorFlow สำหรับ GPU ตอนนี้เวอร์ชั่น 1.4 สนับสนุน CUDA 8.0 และ  cuDNN 6.0 ให้ติดตั้งตามนี้ก่อน

1. เมื่อสำเร็จแล้วติดตั้ง TensorFlow ตามนี้
$sudo apt-get install libcupti-dev
$pip3 install tensorflow-gpu

2. ทดสอบการติดตั้งโดยการรันโปรแกรมง่ายๆ ดังนี้
# Python
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

No comments:

Post a Comment