Tensorflow - cài đặt trên Docker

Docker luôn là giải pháp đóng gói và triển khai các ứng dụng 1 cách nhanh chóng và an toàn. Sau mình xin hướng dẫn các cài đặt Tensorflow trong môi trường container của Docker.

Cài đặt Docker

Mình sử dụng Ubuntu 15.10, bạn có thể xem cách cài đặt Docker cho từng loại hệ điều hành treentrang của Docker: https://docs.docker.com/engine/installation/ubuntulinux/

Launch a Docker

Google cung cấp 4 bản Docker của Tensorflow:

• b.gcr.io/tensorflow/tensorflow: TensorFlow CPU binary image.
• b.gcr.io/tensorflow/tensorflow:latest-devel: CPU Binary image plus source code.
• b.gcr.io/tensorflow/tensorflow:latest-gpu: TensorFlow GPU binary image.
• b.gcr.io/tensorflow/tensorflow:latest-devel-gpu: GPU Binary image plus source code.

Lựa chọn bản Docker phù hợp với nhu cầu sử dụng.

Sau khi cài đặt và khởi động Docker container. Chạy lệnh sau để tự động cài đặt môi trường Tensorflow:

$ docker run -it b.gcr.io/tensorflow/tensorflow

Nếu sử dụng Tensorflow có hỗ trợ GPU, cần phải thêm các tham số để Docker thiết lập GPU, cái này hơi phức tạp, nên Google có viết sẵn 1 script ở đây, chỉ cần tải về và sử dụng, câu lệnh sẽ là:

wget https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/docker/docker_run_gpu.sh
chmod +x ./docker_run_gpu.sh
./docker_run_gpu.sh b.gcr.io/tensorflow/tensorflow:gpu

Kiểm tra cài đặt

Vui lòng xem phần kiểm tra cài đặt trong bài viết sau.