Hướng dẫn cài đặt Tensorflow GPU trên Ubuntu 17.04

Giới thiệu

Chào mọi người, sau đây mình sẽ hướng dẫn cách cài đặt Tensorflow phiên bản GPU trên Ubuntu. Phần lớn các bài hướng dẫn chi tiết cài đặt Tensorflow GPU đã lỗi thời, khi thực hiện theo sẽ gặp lỗi. Sau khi cài thành công mình quyết định ghi chú ngắn gọn các bước cài đặt.

Cài đặt

1. Cài đặt Cuda:

    Vào link: https://developer.nvidia.com/cuda-downloads

    Chọn nền tảng tương ứng, download và chạy các lệnh như hướng dẫn để cài đặt.

2. Tiếp tục thực hiện bước 2 và 3 như trong link sau: 

http://www.nvidia.com/object/gpu-accelerated-applications-tensorflow-installation.html

Note: Ở thời điểm hiện tại, trong bước 2 các bạn tải cuDNN v5.1

3. Từ bước 4 trở đi mình không cần phải làm theo hướng dẫn đó nữa, lúc đó ta sẽ làm như sau:
(Python 2.7;  GPU)
$ pip install tensorflow-gpu
$ sudo nano ~/.bashrc
Thêm dòng sau vào cuối file: 
LD_LIBRARY_PATH=/usr/local/cuda/lib64
Và lưu lại: Ctrl + X --> y --> Enter

(Tránh lỗi không tìm thấy thư viện)
$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

(Test thử)
$ python
>>> import tensorflow
Note: nếu không hiện thông báo lỗi gì tức bạn đã cài thành công. Chúc mừng nhé!

Tham khảo

• http://www.nvidia.com/object/gpu-accelerated-applications-tensorflow-installation.html
• https://www.tensorflow.org/install/install_linux