ติดตั้ง 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))

ติดตั้ง CUDA Toolkit 8.0 และ cuDNN 6.0 บน Ubuntu 16.04 LTS

จากการแนะนำของบลอกนี้ และคู่มือจากเวบของ NVIDIA

1. ดาวน์โหลด CUDA Toolkit 8.0 GA2 จาก https://developer.nvidia.com/cuda-toolkit-archive โดยเลือกไฟล์ตามระบบปฏิบัติการที่ต้องการ สุดท้ายเลือก Installer Type แบบ deb (network)

2. คำสั่งติดตั้งดังนี้

$sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
$sudo apt-get update
$sudo apt-get install cuda-8-0 (จากระทู้นี้)

3. edit ตัวแปร environment

$sudo nano /etc/environment (แล้วเติม :/usr/local/cuda-8.0/bin)
$source /etc/environment (ถ้าไม่เวิร์คให้รีบูต)
$nvcc -V (เพื่อทดสอบ)

4. ดาวน์โหลด cuDNN 6.0 for CUDA 8.0 จาก https://developer.nvidia.com/rdp/cudnn-download โดยต้องเป็นสมาชิกก่อน โดยเลือกโหลดดังนี้ cuDNN v6.0 Runtime Library for Ubuntu16.04 (Deb) และ cuDNN v6.0 Developer Library for Ubuntu16.04 (Deb)

5. คำสั่งติดตั้ง

$sudo dpkg -i .deb (โดยให้ติดตั้ง Runtime Library ก่อนแล้วตามด้วย Developer Library)