KAIKETSU Developer's Diary

株式会社KAIKETSU エンジニアチームのブログです

どうしてもCentOS6でTensorflow x Pythonを使いたいとき

背景

現行サービスとかの関係でどうしてもCentOS6を手放せないけど、同じマシンでTensorflow (keras)をPythonから動かしたかった。

解決策

Dockerを使う

注意点

RHEL6はDockerのサポート外なのでトラブっても放置されます。また、Docker 1.8以降は利用できないため、1.7以前のバージョンを動かすことになります。

インストール

$ yum install https://get.docker.com/rpm/1.7.1/centos-6/RPMS/x86_64/docker-engine-1.7.1-1.el6.x86_64.rpm

以上。

使ってみる

Dockerfile

FROM ubuntu:16.04

RUN apt update && apt install python3 python3-pip && pip3 install --upgrade pip && pip3 install tensorflow keras sklearn h5py
CMD /bin/bash

大体こんな感じです。用途に合わせてpipで入れるモジュールは編集するといいです。

イメージ作成

docker build -t 'tf/ubuntu:16.04' .

コンテナ作成

docker run --name tf -i -t tf/ubuntu:16.04

以上になります。

Dockerの使い方についてはかなりの量ネットに上がっているので割愛しました。

まとめ

ありがとう仮想化時代