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